Python – 从 Pandas 数据框中选择多列

假设以下为 CSV 文件在 Microsoft Excel 中的内容 –

首先，将数据从 CSV 文件中加载到 Pandas 数据框中 –

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

要选择多列记录，请使用方括号。在方括号中提及列，并从整个数据集中获取多个列 –

dataFrame[['Reg_Price','Units']]

例子

以下是代码 –

import pandas as pd

# Load data from a CSV file into a Pandas DataFrame:
dataFrame = pd.read_csv("C:\Users\amit_\Desktop\SalesData.csv")
print("\n读取 CSV 文件...\n",dataFrame)

# 显示两列
res = dataFrame[['Reg_Price','Units']];
print("\n显示两列：\n",res)

输出

这将产生以下输出 –

读取 CSV 文件...
       车   Reg_Price   Units
0      宝马        2500     100
1    雷克萨斯      3500      80
2     奥迪         2500     120
3   捷豹           2000      70
4   野马          2500     110

显示两列：
    Reg_Price   Units
0        2500     100
1        3500      80
2        2500     120
3        2000      70
4        2500     110