pandas - 使用 pandas 中其他列的值名称创建新列

Question

我有一个如下所示的 DataFrame。

DF =

id  w  R
1   A  L    
2   B  J      
3   C  L,J

我现在想要创建一个新列，用于显示列 R 中的值是否出现在行中。

DF2 =

id  w  R    L  J 
1   A  L    1  0 
2   B  J    0  1 
3   C  L,J  1  1 

我尝试了这一行，但结果不是我想要的:

for x in DF.R.unique():
  DF[x]=(DF.R==x).astype(int)


DF2 =

id  w  R    L  J L,J
1   A  L    1  0  0
2   B  J    0  1  0
3   C  L,J  0  0  1

需要什么来解决这个问题？ DF 也很大，慢速方法不起作用。

Answer 1

您需要指定 sep ，在您的示例中是 ,

df.R.str.get_dummies(sep=',')
Out[192]: 
   J  L
0  0  1
1  1  0
2  1  1