python Pandas : How to create a binary matrix from column of lists

我有一个如下所示的 Python Pandas DataFrame:

      1 
0  a, b 
1     c 
2     d 
3     e 

a, b 是表示用户特征列表的字符串

如何将其转换为用户特征的二进制矩阵，如下所示:

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

我看到一个类似的问题Creating boolean matrix from one column with pandas但该列不包含列表条目。

我已经尝试过这些方法，有没有办法将两者合并:

pd.get_dummies()

pd.get_dummies(df[1]) 
 
 
   a, b  c  d  e 
0     1  0  0  0 
1     0  1  0  0 
2     0  0  1  0 
3     0  0  0  1 

df[1].apply(lambda x: pd.Series(x.split()))

      1 
0  a, b 
1     c 
2     d 
3     e 

也对创建这种类型的二进制矩阵的不同方法感兴趣!

感谢任何帮助!

谢谢

请您参考如下方法:

我认为你可以使用:

df = df.iloc[:,0].str.split(', ', expand=True) 
       .stack() 
       .reset_index(drop=True) 
       .str.get_dummies() 
 
print df 
   a  b  c  d  e 
0  1  0  0  0  0 
1  0  1  0  0  0 
2  0  0  1  0  0 
3  0  0  0  1  0 
4  0  0  0  0  1 

编辑:

print df.iloc[:,0].str.replace(' ','').str.get_dummies(sep=',') 
   a  b  c  d  e 
0  1  1  0  0  0 
1  0  0  1  0  0 
2  0  0  0  1  0 
3  0  0  0  0  1