Python Pandas - Tạo DataFrame con bằng toán tử chỉ mục (Indexing Operator)

Toán tử chỉ mục (cặp ngoặc vuông) dùng để tạo DataFrame con. Trước tiên tạo DataFrame 3 cột:

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

Tạo tập con

Một cột đơn lẻ

dataFrame[['Product']]

Nhiều cột

dataFrame[['Opening_Stock', 'Closing_Stock']]

Ví dụ hoàn chỉnh

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 một cột:\n", dataFrame[['Product']])
print("\nTập con nhiều cột:\n", dataFrame[['Opening_Stock', 'Closing_Stock']])

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 một cột:
     Product
0    SmartTV
1  ChromeCast
2    Speaker
3   Earphone

Tập con nhiều cột:
   Opening_Stock  Closing_Stock
0            300            200
1            700            500
2           1200           1000
3           1500            900

Lưu ý: Dùng hai cặp ngoặc vuông [['col']] trả về DataFrame. Dùng một cặp ['col'] trả về Series.