Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Python - Ép kiểu Pandas thành Set (tập hợp)


Để ép kiểu (typecast) pandas thành Set, sử dụng set(). Trước tiên, hãy tạo một DataFrame:

dataFrame = pd.DataFrame({
    "EmpName": ['John', 'Ted', 'Jacob', 'Scarlett', 'Ami', 'Ted', 'Scarlett'],
    "Zone": ['North', 'South', 'South', 'East', 'West', 'East', 'North']
})

Ép kiểu pandas thành set và lấy hợp tập hợp (union):

set(dataFrame.EmpName) | set(dataFrame.Zone)

Ví dụ

Dưới đây là mã hoàn chỉnh:

import pandas as pd
# Tạo DataFrame
dataFrame = pd.DataFrame({
    "EmpName": ['John', 'Ted', 'Jacob', 'Scarlett', 'Ami', 'Ted', 'Scarlett'],
    "Zone": ['North', 'South', 'South', 'East', 'West', 'East', 'North']
})
print("DataFrame ...\n", dataFrame)
# Pandas thành Set
print("\nTypecast Pandas into Set...\n", set(dataFrame.EmpName) | set(dataFrame.Zone))

Kết quả

Điều này sẽ tạo ra đầu ra sau:

DataFrame ...
  EmpName   Zone
0    John  North
1     Ted  South
2   Jacob  South
3 Scarlett   East
4     Ami   West
5     Ted   East
6 Scarlett  North
Typecast Pandas into Set...
set(['Ami', 'East', 'North', 'West', 'Ted', 'South', 'Jacob', 'John', 'Scarlett'])