转换集到数据帧

Question

如何将一组类别转换为 DataFrame？

例如：

A = [{'a', 'c'}, {'a', 'b'}, {'b', 'd'}, {'e'}]

到：

    'a', 'b', 'c', 'd', 'e'
1    1 ,  0 ,  1 ,  0 ,  0 
2    1 ,  1 ,  0 ,  0 ,  0 
3    0 ,  1 ,  0 ,  1 ,  0 
4    0 ,  0 ,  0 ,  0 ,  1  

Answer 1

让我们尝试explode那么crosstab：

s = pd.Series(A).explode()
pd.crosstab(s.index, s)

输出：

col_0  a  b  c  d  e
row_0               
0      1  0  1  0  0
1      1  1  0  0  0
2      0  1  0  1  0
3      0  0  0  0  1

选项 2：get_dummies在explode：

pd.get_dummies(pd.Series(A).explode()).sum(level=0)

输出：

   a  b  c  d  e
0  1  0  1  0  0
1  1  1  0  0  0
2  0  1  0  1  0
3  0  0  0  0  1