Computer >> Máy Tính >  >> Lập trình >> Python

Python Pandas - Tập hợp con DataFrame theo tên cột

Để tạo một tập hợp con DataFrame theo tên cột, hãy sử dụng dấu ngoặc vuông. Sử dụng DataFrame với dấu ngoặc vuông (toán tử lập chỉ mục) và tên cột cụ thể như sau -

dataFrame[‘column_name’]

Đầu tiên, hãy nhập thư viện được yêu cầu với bí danh -

import pandas as pd

Tạo Pandas DataFrame với các bản ghi Sản phẩm -

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

Hãy để chúng tôi tìm nạp một tập hợp con, tức là chúng tôi chỉ tìm nạp các bản ghi cột Sản phẩm

dataFrame['Product']

Ví dụ

Sau đây là mã

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"\nDisplaying a subset:\n",dataFrame['Product']

Đầu ra

Điều này sẽ tạo ra kết quả sau

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

Displaying a subset:
0      SmartTV
1      ChromeCast
2      Speaker
3      Earphone
Name: Product, dtype: object