Python – 如何从Pandas DataFrame选择一列

要从DataFrame中选择一列，只需使用方括号获取。在括号中提及要选择的列，就可以了，例如

dataFrame[‘ColumnName’]

首先，导入所需的库――

import pandas as pd

现在，创建一个DataFrame。我们有两列――

dataFrame = pd.DataFrame(
   {
      "Car": ['宝马', '雷克萨斯', '奥迪', '野马', '宾利', '捷豹'],
      "Units": [100, 150, 110, 80, 110, 90]
   }
)

要选择单个列，请使用方括号注明列名，如下所示。这里，我们的列名是 ‘Car’ –

dataFrame ['Car']

更多Pandas文章，请阅读：Pandas教程

示例

以下是代码――

import pandas as pd

# Create DataFrame
dataFrame = pd.DataFrame(
   {
      "Car": ['宝马', '雷克萨斯', '奥迪', '野马', '宾利', '捷豹'],
      "Units": [100, 150, 110, 80, 110, 90]
   }
)

print"DataFrame ...\n",dataFrame

# selecting a column
print"\n选取并显示一列 = \n",dataFrame ['Car']

输出

这将产生以下输出――

DataFrame ...
       Car   Units
0      宝马     100
1    雷克萨斯    150
2     奥迪     110
3  野马         80
4  宾利        110
5   捷豹        90

选取并显示一列 =
0      宝马
1    雷克萨斯
2     奥迪
3    野马
4    宾利
5     捷豹
Name: Car, dtype: object