python - 如何在 Pandas 中转换分类数据框

Question

我的数据框在下面

Sex type 
M    A  
W    B
W    A
M    C

我想得到以下结果

M W A B C
1 0 1 0 0
0 1 0 1 0
0 1 1 0 0
1 0 0 0 1

我试过了

df['性别','类型'].get_dummies()

但效果不佳。

我怎样才能得到这个结果？

Answer 1

看来你需要pandas.get_dummies :

print (pd.get_dummies(df, prefix_sep='', prefix=''))
   M  W  A  B  C
0  1  0  1  0  0
1  0  1  0  1  0
2  0  1  1  0  0
3  1  0  0  0  1

此外，如果列名中的一些重复项是必要的，则按列 axis=1 和 level=0 进行 groupby 并汇总 max:

print (pd.get_dummies(df, prefix_sep='', prefix='').groupby(axis=1, level=0).max())
   A  B  C  M  W
0  1  0  0  1  0
1  0  1  0  0  1
2  1  0  0  0  1
3  0  0  1  1  0