python - 如何在 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

聚合和排序调用的结果是一个 pandas 系列，其索引是您要查找的组。因此，要找到大于某个 cutOffvalue

的组

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

当然，如果您希望将其作为列表或集合，您可以轻松地转换最终值

groups = list(groups)