python - 如何根据 Pandas 中一系列列的特定条件选择行

Question

假设，我有以下数据框:

A B C D E F
1 1 1 0 0 0
0 0 0 0 0 0
1 1 0.9 1 0 0
0 0 0 0 -1.95 0
0 0 0 0 2.75 0
1 1 1 1 1 1

我想根据 C、D、E 和 F 列选择只有 0 和 1 (0 & 1) 的行。对于这个例子，预期的输出是

 A B C D E F
 1 1 1 0 0 0

我如何考虑 pandas 中的一系列列来做到这一点？

提前致谢。

Answer 1

让我们尝试使用 loc 进行 bool 索引来过滤行:

c = ['C', 'D', 'E', 'F']
df.loc[df[c].isin([0, 1]).all(1) & df[c].eq(0).any(1) & df[c].eq(1).any(1)]

结果:

   A  B    C  D    E  F
0  1  1  1.0  0  0.0  0