如何用Pandas实现多组条件筛选Python DataFrame?
解决多组条件筛选DataFrame的问题
你的代码存在两个关键错误:
- 两组条件之间误用逗号分隔,Pandas筛选中需用**逻辑或运算符
|**连接不同条件组 - 未将每组需同时满足的条件用括号整体包裹,会导致逻辑运算优先级混乱
正确写法(基础版)
把每组条件分别用括号括起,再用|连接:
dfy = dfx[ ((dfx['material']=='stone') & (dfx['waterbody']=='marsh') & (dfx['mosscat']=='average')) | ((dfx['material']=='brick') & (dfx['waterbody']=='river') & (dfx['mosscat']=='average')) ]
更简洁的写法(使用query)
用query()方法可以写出更直观的筛选逻辑:
dfy = dfx.query( "(material == 'stone' and waterbody == 'marsh' and mosscat == 'average') or " "(material == 'brick' and waterbody == 'river' and mosscat == 'average')" )
进一步简化逻辑
因为两组条件中mosscat均为'average',可以先筛选这个共性条件,再处理剩余组合:
dfy = dfx[dfx['mosscat'] == 'average'].query( "(material == 'stone' and waterbody == 'marsh') or (material == 'brick' and waterbody == 'river')" )
内容的提问来源于stack exchange,提问作者data_life
相关产品推荐
相关产品推荐

