Skip to main content
 首页 » 编程设计

python之在同一行过滤来自 pandas.groupby 的结果

2025年05月04日13mfryf

如何在一行中根据 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