如何在pandas DataFrame中实现排除无关条目的多条件过滤
错误原因
pandas布尔索引规则中,&(按位与)的运算优先级高于|(按位或)。原始代码未给或逻辑添加括号,实际执行逻辑为(Edition == 2004 且 Discipline == "Athletics") 或 Discipline == "Aquatics",因此会返回所有年份下Discipline为Aquatics的无关数据。
修正写法
给Discipline字段的两个或判断整体加上括号,确保整体逻辑是「Edition固定为2004,且Discipline为Athletics或Aquatics」:
df[(df.Edition == 2004) & ((df.Discipline == "Athletics") | (df.Discipline == "Aquatics"))]
简化写法
同字段匹配多个值时,优先使用isin()方法,代码可读性更高,也不会出现运算符优先级导致的逻辑错误:
df[(df.Edition == 2004) & df.Discipline.isin(["Athletics", "Aquatics"])]
内容的提问来源于stack exchange,提问作者shrey_shankar
相关产品推荐
相关产品推荐

