itertools.groupby() ne regroupez pas correctement

J'ai ces données:

self.data = [(1, 1, 5.0),
             (1, 2, 3.0),
             (1, 3, 4.0),
             (2, 1, 4.0),
             (2, 2, 2.0)]

Lorsque j'exécute ce code:

for mid, group in itertools.groupby(self.data, key=operator.itemgetter(0)):

pour list(group) j'obtiens:

[(1, 1, 5.0),
 (1, 2, 3.0),
 (1, 3, 4.0)]

qui est ce que je veux.

Mais si j'utilise 1 au lieu de 0

for mid, group in itertools.groupby(self.data, key=operator.itemgetter(1)):

de groupe par le second nombre de tuples, je ne reçois:

[(1, 1, 5.0)]

même si il y a d'autres tuples qui ont "1" 1 (2e) position.

InformationsquelleAutor user994165 | 2011-11-14

itertools python

21

itertools.groupby recueille contiguë éléments avec la même clé.
Si vous souhaitez que tous les éléments avec la même clé, vous devez trier self.data premier.
```
for mid, group in itertools.groupby(
    sorted(self.data,key=operator.itemgetter(1)), key=operator.itemgetter(1)):
```
- J'avais trié précédemment sur la position zéro. Je viens donc de nouveau triées avant de faire le groupby et il fonctionne. auto.les données.sort(key=opérateur.itemgetter(1))
InformationsquelleAutor unutbu
19

Variante sans tri (par dictionnaire). Devrait être de mieux en terme de performance.
```
def full_group_by(l, key=lambda x: x):
    d = defaultdict(list)
    for item in l:
        d[key(item)].append(item)
    return d.items()
```
- Revint se poster la même chose, je n'avais pas lu votre réponse! C'est clairement la voie à suivre 🙂
- Malheureusement, les clés de toutes les hashable puis, de sorte qu'il ne fonctionne pas si ceux-ci sont pour des exemples de listes, contrairement à itertools.groupby...
InformationsquelleAutor Kostiantyn Rybnikov
2

Ci-dessous "fixe" plusieurs ennuis avec Python itertools.groupby.
```
def groupby2(l, key=lambda x:x, val=lambda x:x, agg=lambda x:x, sort=True):
    if sort:
        l = sorted(l, key=key)
    return ((k, agg((val(x) for x in v))) \
        for k,v in itertools.groupby(l, key=key))
```
Plus précisément,
1. Il ne nécessite pas que vous triez vos données.
2. Il ne nécessite pas que vous devez utiliser key comme paramètre nommé seulement.
3. La sortie est propre générateur de tuple(key, grouped_values) où les valeurs sont spécifiées par le 3e paramètre.
4. Capacité à appliquer des fonctions d'agrégation comme la somme ou avg facilement.
Exemple D'Utilisation
```
import itertools
from operator import itemgetter
from statistics import *

t = [('a',1), ('b',2), ('a',3)]
for k,v in groupby2(t, itemgetter(0), itemgetter(1), sum):
  print(k, v)
```
Ce estampes,
```
a 4
b 2
```
Jouer avec ce code

InformationsquelleAutor Shital Shah

Vous devez vous connecter pour publier un commentaire.