如何在一行中根据 groupby 转换的输出过滤数据框。
到目前为止,我得到了以下有效的方法,但我想知道是否有更简单/更有效的方法。
import pandas as pd
df = pd.DataFrame({'A': ['one', 'one', 'two', 'two'], 'B': [1, 1, 6, 6]})
df.groupby('A').filter(lambda x: x['B'].sum()>10).groupby('A').B.sum()
提前致谢。
请您参考如下方法:
这是一种方式
df.groupby('A').sum().loc[lambda x : x['B']>10]
Out[251]:
B
A
two 12
