Python - Cách chọn tập con Pandas DataFrame

File CSV SalesData.csv.

Tải dữ liệu:

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

Chọn tập con cột đơn dùng ngoặc vuông:

dataFrame['Car']

Ví dụ

import pandas as pd 
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesData.csv") 
print("\nĐọc file CSV...\n", dataFrame) 
res1 = dataFrame['Car'] 
print("\nCột Car:\n", res1) 
res2 = dataFrame[['Car', 'Units']] 
print("\n2 cột:\n", res2)

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 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