Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Python - Chọn nhiều cột từ Pandas DataFrame


File CSV SalesData.csv.

Tải dữ liệu:

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

Chọn nhiều cột dùng ngoặc vuông:

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

Ví dụ

import pandas as pd 
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesData.csv") 
print("\nĐọc file CSV...\n", dataFrame) 
res = dataFrame[['Reg_Price', 'Units']] 
print("\nHiển thị 2 cột:\n", res)

Kết quả

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 two columns: 
  Reg_Price  Units 
0 2500  100 
1 3500  80 
2 2500  120 
3 2000  70 
4 2500  110