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

更新日期:2021 年 9 月 14 日

1K+ 浏览次数

开启你的 职业生涯

完成课程以获得认证

开始
广告
© . All rights reserved.