Python - Tạo một tập hợp con của các cột bằng cách sử dụng filter ()

Để tạo một tập hợp con các cột, chúng ta có thể sử dụng filter (). Thông qua đó, chúng ta có thể lọc các giá trị cột có mẫu tương tự bằng cách sử dụng toán tử like. Đầu tiên, hãy tạo DataFrame với 3 cột -

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

Bây giờ, chúng ta hãy tạo một tập hợp con có nhiều cột -

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

Tạo một tập hợp con với các tên có mẫu tương tự -

dataFrame.filter(like='Open')

Ví dụ

Sau đây là mã 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"\nDisplaying a subset using indexing operator:\n",dataFrame[['Product']]

print"\nDisplaying a subset with multiple columns:\n",dataFrame[['Opening_Stock','Closing_Stock']]

print"\nDisplaying a subset with similarly patterned names:\n",dataFrame.filter(like='Open')

Đầ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 using indexing operator:
      Product
0     SmartTV
1  ChromeCast
2     Speaker
3    Earphone

Displaying a subset with multiple columns:
   Opening_Stock   Closing_Stock
0         300             200
1         700             500
2        1200            1000
3        1500             900

Displaying a subset with similarly patterned names:
   Opening_Stock
0          300
1          700
2       1200
3 1500