python - 如何对 pandas Dataframe 的一列中的所有值进行集合并集？

Question

数据帧的前两行，df:

0|50331648|{1,2,3,4,5}|6  
1|50331649|{3,5,7,8}|2  

执行操作后，我只需要一个包含{1,2,3,4,5,7,8}的集合。

如何实现？

Answer 1

假设 "B" 是正在考虑的列名称，您可以使用 set.union在得到的解包列表上:

set.union(*df['B'].tolist())
{1, 2, 3, 4, 5, 7, 8}

(或)

将这些作为可调用函数提供给 reduce:

from functools import reduce      # If you're on Py3k
reduce(set.union, df['B'].tolist())
{1, 2, 3, 4, 5, 7, 8}

数据:

df = pd.DataFrame(dict(A=[50331648, 50331649],
                       B=[{1,2,3,4,5}, {3,5,7,8}],
                       C=[6,2])
                 )