Python - 如何选择 Pandas DataFrame 的子集

假设以下是在 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("\n仅显示一列 Car :\n",res1)

# 显示两列
res2 = dataFrame[['Car','Units']];
# 显示另一个子集
print("\n显示两列 Car :\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

仅显示一列 Car :
0      BMW
1    Lexus
2     Audi
3   Jaguar
4  Mustang
Name: Car, dtype: object

显示两列:
       Car   Units
0      BMW     100
1    Lexus      80
2     Audi     120
3   Jaguar      70
4  Mustang     110

相关文章

• Python 中通过 K 扩展列表
• Python - 使用交替的 0 和 1 初始化列表
• Python 中元组列表到字典的转换
• Python - 从两个列表中配对，使得成对的元素不相同
• 使用 Python 和 SQLite 的 SQL
• 使用 pytrie 模块在 Python 中进行前缀匹配
• 创建和访问 Python 包
• 如何使用正则表达式和数据类型选择多个 DataFrame 列
• 如何在 Pandas 中查找和过滤重复行?
• 用 Python 编写程序以正确的顺序查找机场?

有用资源

• python 参考教程 - 该教程包含有关 python 的更多信息:https://www.cainiaomax.com/python/