Python Pandas - Tạo tập con DataFrame theo tên cột

Để tạo tập con DataFrame theo tên cột, dùng cặp ngoặc vuông (toán tử chỉ mục) với tên cột cụ thể: dataFrame['tên_cột'].

Ví dụ

import pandas as pd

dataFrame = pd.DataFrame({
    "Product": ["SmartTV", "ChromeCast", "Speaker", "Earphone"],
    "Opening_Stock": [300, 700, 1200, 1500],
    "Closing_Stock": [200, 500, 1000, 900]
})

print("DataFrame...\n", dataFrame)
print("\nTập con cột Product:\n", dataFrame['Product'])

Kết quả đầu ra

DataFrame...
  Closing_Stock  Opening_Stock    Product
0           200            300    SmartTV
1           500            700  ChromeCast
2          1000           1200    Speaker
3           900           1500   Earphone

Tập con cột Product:
0    SmartTV
1  ChromeCast
2    Speaker
3   Earphone
Name: Product, dtype: object

Lưu ý:

• df['col'] → trả về Series.
• df[['col']] (hai ngoặc) → trả về DataFrame 1 cột.
• df[['col1', 'col2']] → DataFrame nhiều cột.