python - Pandas 根据条件进行过滤分组

Question

我有一个与此处提到的非常相似的数据集 http://pandas.pydata.org/pandas-docs/stable/10min.html#grouping

>>> df
     A      B         C
0  foo    one -1.735400
1  bar    one -0.148954
2  foo    two  0.103798
3  bar  three -0.576249
4  foo    two  1.379046
5  bar    two  0.802281
6  foo    one -0.758771
7  foo  three  1.270179

我已执行命令 df.groupby(['A','B'])['C'].sum()

>>> grpd = df.groupby(['A','B'])['C'].sum()

>>> grpd
A    B    
bar  one     -0.148954
     three   -0.576249
     two      0.802281
foo  one     -2.494171
     three    1.270179
     two      1.482844
Name: C, dtype: float64

在我的特定情况下，A 列有 storeid，B 是月份(全年)，C 是销售值(value)

目标是找到每个月销量超过特定值的所有 storeid。

示例问题 - 哪个故事的任何一个月的销量超过 1000 美元？

我可以轻松地在 SQL 中执行此操作，但不确定如何在 Pandas 数据框架中执行此操作。

Answer 1

如果我理解正确的话，你可以groupby MuliIndex 级别，然后使用过滤。

grpd.groupby(level='A').filter(lambda grp: (grp > 1000).all())