python - 丢弃 Pandas 分组操作中的分组

对于以下数据帧（df），我正在执行以下操作：

import pandas as pd
def fun(group):
    if(group.A.min() > 0.0):
        # discard the group
        return
    else:
        return group

df.groupby('cokey').apply(fun)

cokey       A   B
11168155    0   18
11168155    18  56
11168155    56  96
11168155    96  152
11168324    76  86
11168324    86  152

大熊猫在groupby操作期间是否有更好的方法丢弃一组

最佳答案

只需应用filter：

In [7]:

df.groupby('cokey')[['A','B']].filter(lambda x: x['A'].min() > 50)
Out[7]:
    A    B
4  76   86
5  86  152

因此，就您而言：

df.groupby('cokey')[['A','B']].filter(lambda x: x['A'].min() <= 0)