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

Question

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

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

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

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