开发者社区

文档建议反馈控制台

文章/答案/技术大牛

发布

社区首页 >问答首页 >在pandas groupby中查找满足条件的组

问在pandas groupby中查找满足条件的组
EN

Stack Overflow用户

提问于 2019-03-20 22:05:56

回答 3查看 1.7K关注 0票数 3

这是我的数据框架：

df = pd.DataFrame({'a':list('xxxyyzz'), 'b':[10,20,30,5,3,1,2]})

我对它们进行分组：

groups = df.groupby('a')

我想打印至少有一个b大于20的组。在这种情况下，我想打印x。这是我想要的结果：

EN

回答 3

Stack Overflow用户

回答已采纳

发布于 2019-03-20 22:07:25

按Series.gt比较值，按a列分组，如Series - df['a']，并使用GroupBy.transform和GroupBy.any测试每个组至少一个True：

df1 = df[df['b'].gt(20).groupby(df['a']).transform('any')]
print (df1)
   a   b
0  x  10
1  x  20
2  x  30

票数 5

EN

Stack Overflow用户

发布于 2019-03-20 22:07:31

您可以使用any检查哪些值高于20、GroupBy列a和transform，以便只选择那些至少有一行满足条件的组：

df[df.b.gt(20).groupby(df.a).transform('any')]

   a   b
0  x  10
1  x  20
2  x  30

票数 4

EN

Stack Overflow用户

发布于 2019-03-20 22:07:41

不需要groupby，只需执行isin即可

df[df.a.isin(df.loc[df.b>20,'a'])]
Out[996]: 
   a   b
0  x  10
1  x  20
2  x  30

票数 4

EN

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/55262665

复制

相似问题