我有以下数据帧:
A | B | C | D
1 | 2 | 4 | 5
0 | 2 | 5 | 2
1 | 1 | 2 | 1
我想要一个新的DataFrame,它只取列B, C
,但只取其中的A == 1
。这就是我想要的:
B | C
2 | 4
1 | 2
虽然我可以先过滤条件中的A == 1
,然后从结果中只取列B, C
,但却是一个更"大"的条件;蟒蛇;怎么做?
在列表中使用带掩码和列名的DataFrame.loc
:
df1 = df.loc[df.A == 1, ['B','C']]