Wie kann ich die Gruppen mit der Größe mehr als einen Wert in Python finden?

Question

Ich lese die Daten in einen DataFrame und nannte sie Daten. Ich habe die folgende Abfrage in Python:

data[data["gender"]=="male"].groupby('age').city.nunique().sort_values(ascending=False) 


age 
29 86 
24 85 
21 81 
25 81 
20 81 
28 78 
27 78 

jetzt mag ich die jene Gruppen finden Größe ist mehr als 80. Wie kann ich das tut in Python?

Answer 1

Das Ergebnis Ihres Aggregations- und Sortieraufrufs ist eine Pandas-Serie, deren Index die Gruppen sind, nach denen Sie suchen. So finden Sie die Gruppen mit mehr als einer bestimmten cutOffvalue

cutOffValue = 80 
counts = data[data["gender"]=="male"].groupby('age').city.nunique().sort_values(ascending=False) 
groups = counts[counts > cutOffValue].index 

Und natürlich, wenn Sie es als Liste oder einstellen möchten, können Sie leicht den Endwert werfen

groups = list(groups)