Python - 如何从 Pandas DataFrame 中选择一列

要从 DataFrame 中选择一列，只需使用方括号获取它。在括号中提到要选择的列，就是这样，例如

dataFrame[‘ColumnName’]

首先，导入所需的库 -

import pandas as pd

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

dataFrame = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
      "Units": [100, 150, 110, 80, 110, 90]
   }
)

要仅选择单个列，请使用方括号提及列名称，如下所示。在这里，我们的列名称是“汽车” -

dataFrame ['Car']

示例

以下是代码 -

import pandas as pd

# Create DataFrame
dataFrame = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
      "Units": [100, 150, 110, 80, 110, 90]
   }
)

print"DataFrame ...\n",dataFrame

# selecting a column
print"\nSelecting and displaying only a single column = \n",dataFrame ['Car']

这将产生以下输出 -

DataFrame ...
       Car   Units
0      BMW     100
1    Lexus     150
2     Audi     110
3  Mustang      80
4  Bentley     110
5   Jaguar      90

Selecting and displaying only a single column =
0      BMW
1    Lexus
2     Audi
3  Mustang
4  Bentley
5   Jaguar
Name: Car, dtype: object