python - Pandas 高效过滤 : Same filter condition on multiple columns

Question

假设我有以下数据:

df = pd.DataFrame({'col1': [1, 2, 1],
                   'col2': [2, 4, 3],
                   'col3': [3, 6, 5],
                   'col4': [4, 8, 7]})

有没有一种方法可以使用列表理解来有效地过滤数据？例如，如果我想找到所有 col2 为偶数或 col3 为偶数或 col 4 为偶数的情况，是否有比只写这个更简单的方法？

df[(df['col2'] % 2 == 0) | (df['col3'] % 2 == 0) | (df['col4'] % 2 == 0)]

如果我可以传入一个列列表和要检查的条件，那就太好了。

Answer 1

df[(df[cols] % 2 == 0).any(axis=1)]

cols 是您的列列表