Python/Pandas 2 个系列的元素明智并集，每个元素中包含集合

Question

我有 2 个 pandas 数据系列，我知道它们的长度相同。每个 Series 在每个元素中都包含 sets() 。我想找出一种计算上有效的方法来获得这两个系列集合的元素明智的联合。我已经创建了一个简化版本的代码，其中包含 fake 和 short Series，以便在下面使用。这种实现是一种非常低效的方法。必须有一种更快的方法来做到这一点。我真正的系列要长得多，我必须做这个操作几十万次。

import pandas as pd

set_series_1 = pd.Series([{1,2,3}, {'a','b'}, {2.3, 5.4}])
set_series_2 = pd.Series([{2,4,7}, {'a','f','g'}, {0.0, 15.6}])

n = set_series_1.shape[0]  
for i in range(0,n):
    set_series_1[i] = set_series_1[i].union(set_series_2[i])

print set_series_1        
>>> set_series_1
0          set([1, 2, 3, 4, 7])
1             set([a, b, g, f])
2    set([0.0, 2.3, 15.6, 5.4])
dtype: object

我已经尝试将系列组合到一个数据框中并使用应用函数，但我收到一条错误消息，指出不支持集合作为数据框元素。

Answer 1

pir4

在测试了几个选项后，我终于想出了一个好的... pir4 下面。

测试

def jed1(s1, s2):
    s = s1.copy()
    n = s1.shape[0]
    for i in range(n):
        s[i] = s2[i].union(s1[i])
    return s

def pir1(s1, s2):
    return pd.Series([item.union(s2[i]) for i, item in enumerate(s1.values)], s1.index)

def pir2(s1, s2):
    return pd.Series([item.union(s2[i]) for i, item in s1.iteritems()], s1.index)

def pir3(s1, s2):
    return s1.apply(list).add(s2.apply(list)).apply(set)

def pir4(s1, s2):
    return pd.Series([set.union(*z) for z in zip(s1, s2)])

Python/Pandas 2 个系列的元素明智并集，每个元素中包含集合

Python/Pandas element wise union of 2 Series containing sets in each element

set

series

python-3.x

pandas

set-union

pir4

测试