python - 有没有办法打破带有类别的 pandas 列，以将类别名称作为列名称来分隔 true 或 false 列

Question

我有一个包含以下列的数据框:

df = pd.DataFrame({"A": [1,2,1,2,2,2,0,1,0]})

我想要:

df2 = pd.DataFrame({"0": [0,0,0,0,0,0,1,0,1],"1": [1,0,1,0,0,0,0,1,0],"2": [0,1,0,1,1,1,0,0,0]}) 

有没有一种优雅的方法可以使用 oneliner 来做到这一点。

注意

• 我可以使用 df['0'] = df['A'].apply(find_zeros) 来做到这一点
• 我不介意“A”是否包含在决赛中。

Answer 1

使用get_dummies :

df2 = pd.get_dummies(df.A)
print (df2)
   0  1  2
0  0  1  0
1  0  0  1
2  0  1  0
3  0  0  1
4  0  0  1
5  0  0  1
6  1  0  0
7  0  1  0
8  1  0  0