我有一些DataFrame,df
我想在其中'columnA'
进行分组,过滤具有的组的成员len >= 3
,并选择具有值Pizza中的成员'columnB'
。
下一行
df.groupby('columnA').filter(lambda x: len(x) >= 3)['columnB'] == 'pizza'
返回带有boolean in的所有行的索引'columnB'
。
这当然起作用:
df2 = df.groupby('columnA').filter(lambda x: len(x) >= 3)
df2 = df2[(df2['columnB'] == 'pizza')]
但是我可以一行完成吗?
@Lambda在评论中提供了解决方案:
df.groupby("columnA").filter(lambda x: len(x)>=3).query('columnB=="pizza"')
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句