如何在python中找到大小大於值的組？

Question

我將數據讀入DataFrame並將其稱為數據。 我在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

現在我想找到那些大小超過80的組。如何在python中做到這一點？

Answer 1

聚合和排序調用的結果是一個熊貓系列，其索引是您要查找的組。 所以要找到大於某個cutOffvalue

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

當然，如果您希望將其作為列表或集合，則可以輕松地轉換最終值

groups = list(groups)