Python: Jak uzyskać długość itertools _grouper

Question

Pracuję z itertools Python i przy użyciu groupby sortowania grupy par przez ostatni element. Zrobiłem to, aby sortować i mogę przejrzeć grupy, ale dobrze byłoby móc uzyskać długość każdej grupy bez konieczności powtarzania każdego z nich, zwiększając licznik.

Projekt jest skupiony w kilka punktów danych. Pracuję z parami (numpy.array, int), gdzie numpy tablicy jest punkt danych i liczba całkowita jest etykietą klastra

Oto mój odpowiedni kod:

data = sorted(data, key=lambda (point, cluster):cluster) 
for cluster,clusterList in itertools.groupby(data, key=lambda (point, cluster):cluster): 
    if len(clusterList) < minLen: 

W ostatnim wierszu "jeśli len (clusterList) < minLen:", pojawia się błąd, że obiekt typu "itertools._grouper" nie ma len().

Sprawdziłem dostępne operacje dla _grouperów, ale nie mogę znaleźć niczego, co wydaje się zapewniać długość grupy.

Answer 1

Tylko dlatego, że nazywasz to clusterList, nie tworzy listy! Jest to w zasadzie leniwy iterator, zwracający każdy element, jaki jest potrzebny. można przekonwertować go na liście, jak ten, choć:

clusterList = list(clusterList) 

Albo zrobić i dostać jego długość w jednym kroku:

length = len(list(clusterList)) 

Jeśli nie chcą zająć pamięć podejmowania to lista, można to zrobić w zamian:

length = sum(1 for x in clusterList) 

należy pamiętać, że oryginalny iterator będą spożywane w całości albo przez przekształcenie go do listy lub z zastosowaniem preparatu sum().

Answer 2

clusterList to iterable, ale nie jest to list. Czasami może to być trochę mylące. Możesz wykonać pętlę for przez clusterList, ale nie możesz robić innych rzeczy na liście (plaster, len itp.). Naprawiono: przypisano wynik list(clusterList) do clusterList.