Python - Cách chọn cột từ DataFrame Pandas

Để chọn một cột từ DataFrame, chỉ cần truy xuất bằng ngoặc vuông với tên cột: dataFrame['TênCột'].

Ví dụ

import pandas as pd

# Tạo DataFrame
dataFrame = pd.DataFrame({
    "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
    "Units": [100, 150, 110, 80, 110, 90]
})

print("DataFrame ...\n", dataFrame)

# Chọn cột Car
print("\nChọn cột Car:\n", dataFrame['Car'])

Kết quả đầu ra

DataFrame ...
       Car  Units
0      BMW    100
1    Lexus    150
2     Audi    110
3  Mustang     80
4  Bentley    110
5   Jaguar     90

Chọn cột Car:
0      BMW
1    Lexus
2     Audi
3  Mustang
4  Bentley
5   Jaguar
Name: Car, dtype: object

Giải thích

• df['col'] trả về Series.
• df[['col']] (hai ngoặc) trả về DataFrame 1 cột.
• df[['col1', 'col2']] trả về DataFrame nhiều cột.
• Cũng có thể dùng attribute: df.Car (chỉ khi tên cột hợp lệ identifier Python).