Python – 如何选择Pandas DataFrame的一个子集

假设我们的CSV文件在Microsoft Excel中打开后的内容如下−

首先，将数据从CSV文件加载到Pandas DataFrame中−

dataFrame = pd.read_csv("C:\Users\amit_\Desktop\SalesData.csv")

要选择一个子集，请使用方括号。在括号中提及列，并从整个数据集中获取单个列−

dataFrame['Car']

示例

以下是代码−

import pandas as pd

#将数据从CSV文件加载到 Pandas DataFrame 中
dataFrame = pd.read_csv("C:\Users\amit_\Desktop\SalesData.csv")
print("\nReading the CSV file...\n",dataFrame)

# 仅显示一列
res1 = dataFrame['Car'];
# 显示一个子集
print("\nDisplaying only one column Car : \n",res1)

# 显示两列
res2 = dataFrame[['Car','Units']];
# 显示另一个子集
print("\nDisplaying two columns : \n",res2)

输出

这会产生以下输出−

Reading the CSV file...
       Car   Reg_Price   Units
0      BMW        2500     100
1    Lexus        3500      80
2     Audi        2500     120
3   Jaguar        2000      70
4  Mustang        2500     110

Displaying only one column Car :
0      BMW
1    Lexus
2     Audi
3   Jaguar
4  Mustang
Name: Car, dtype: object

Displaying two columns :
       Car   Units
0      BMW     100
1    Lexus      80
2     Audi     120
3   Jaguar      70
4  Mustang     110