python - Pandas:为什么在 boolean 索引后需要双括号来选择列

Question

对于像下面这样的 df 表，

   A B C D
0  0 1 1 1
1  2 3 5 7
3  3 1 2 8

为什么在 boolean 索引后选择特定列需要双括号？

the [['A','C']] part of

df[df['A'] < 3][['A','C']]

Answer 1

对于pandas对象(Series、DataFrame)，索引操作符[]只接受

1. colname或选择列的列名列表
2. 切片或 boolean 数组来选择行，即它只引用数据框的一维。

对于 df[[colname(s)]] ，内括号是列表，外括号是索引运算符，即如果选择两列或多列，则必须使用双括号。对于一个列名，单对括号返回一个系列，而双括号返回一个数据框。

另外，df.ix[df['A'] < 3,['A','C']]或 df.loc[df['A'] < 3,['A','C']]在避免返回副本与数据框 View 方面比链式选择要好。

请引用 pandas documentation了解详情