Python – Typecasting Pandas into set

将Pandas转换为Set，使用set()函数。首先，让我们创建一个DataFrame——

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

将Pandas转换为Set，然后获取Set的并集——

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

例子

下面是完整的代码——

import pandas as pd

# 创建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转换为Set
print("\n将Pandas转换为Set...\n", set(dataFrame.EmpName) | set(dataFrame.Zone))

输出

这将产生以下输出——

DataFrame ...
    EmpName   Zone
0      John   North
1       Ted   South
2     Jacob   South
3  Scarlett   East
4       Ami   West
5       Ted   East
6  Scarlett   North

将Pandas转换为Set...
set(['Ami', 'East', 'North', 'West', 'Ted', 'South', 'Jacob', 'John', 'Scarlett'])