python - 在 Pandas 中显示带有条件的两列

Question

假设我有一个数据框df，例如

A  B  C
0  a  1
1  b  1
2  c  2

我想在 C==1 时返回 B 和 C，就像这样

B  C
a  1
b  1

我已经得到了 df.B[df.C==1]，它返回了

B
a
b

这是正确的(逆向)但在切片中是错误的。我如何获得 C？

Answer 1

您可以使用 loc或 query :

print df.loc[df.C==1, ['B','C']]
   B  C
0  a  1
1  b  1

print df[['B','C']].query('C == 1')
   B  C
0  a  1
1  b  1

或者如果您只需要 C 列:

print df.loc[df.C==1, 'C']

0    1
1    1
Name: C, dtype: int64