热线电话:13121318867

登录
2019-03-14 阅读量: 741
如果不满足条件,则丢弃整个组

我有一个pandas数据帧如下:
df = pd.DataFrame({'alpha':['a','a','b','b'],'beta':[1,2,3,4]})

在此输入图像描述


我们的想法是只返回列中满足某些条件的组beta,否则丢弃整个组
我想要的结果是:

在此输入图像描述


解决办法:可以使用groupby.filter如下:

print (df.groupby('alpha').filter(lambda x: (x.beta >1).all()))

alpha beta

2 b 3

3 b 4

对于lambda,我理解你希望根据你的预期输出all,该组的值beta应该超过1。

熊猫 熊猫组织

0.0000
4
关注作者
收藏
评论(0)

发表评论

暂无数据
推荐帖子