Python - 如何选择 Pandas DataFrame 的子集

  • Z时代
  • 2024-01-10
  • 分类：综合

假设以下是我们在 Microsoft Excel 中打开的 CSV 文件的内容 -

首先，将 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

以上是 Python - 如何选择 Pandas DataFrame 的子集 的全部内容， 来源链接： utcz.com/z/363173.html

回到顶部