Python pandas数据帧为什么使用双方括号

Question

为什么

print(data["column"].shape)

打印(1819,)和

print(data[["column']].shape)

打印(1819,1)

Answer 1

data["column"]返回一个Pandas系列，它总是一直是形状（n，），即它总是没有列只有一行。

data[["column"]]返回一个形状为（m，n）

的Pandas DataFrame

如果您想在数据框中使用多个列，可以使用双括号，如下所示。

data[["col1", "col2"]]