如何通过pandas get_dummies（）方法为某些列创建虚拟对象？

Question

df = pd.DataFrame({'A': ['x', 'y', 'x'], 'B': ['z', 'u', 'z'],
                  'C': ['1', '2', '3'],
                  'D':['j', 'l', 'j']})

我只希望A列和D列获得B列的虚拟变量，而不是Bpd.get_dummies(df)列。

我希望最终结果包含所有列，这意味着列C和列B退出，如'A_x','A_y','B','C','D_j','D_l'。

Answer 1

使用带有必需参数的get_dummies（）可以不进行串联

In [294]: pd.get_dummies(df, prefix=['A', 'D'], columns=['A', 'D'])
Out[294]: 
   B  C  A_x  A_y  D_j  D_l
0  z  1  1.0  0.0  1.0  0.0
1  u  2  0.0  1.0  0.0  1.0
2  z  3  1.0  0.0  1.0  0.0