Question

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

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

在群组操作期间，pandas是否有更好的方法来丢弃群组

Answer 1

只需应用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)

通过操作丢弃pandas group中的组

1 个答案: