python - 在几个 DataFrame 列上运行 get_dummies？

Question

如何在多个 DataFrame 列上惯用地运行像 get_dummies 这样的函数，它需要一个列并返回多个列？

Answer 1

使用 pandas 0.19，您可以在一行中完成:

pd.get_dummies(data=df, columns=['A', 'B'])

Columns 指定在何处进行 One Hot Encoding。

>>> df
   A  B  C
0  a  c  1
1  b  c  2
2  a  b  3

>>> pd.get_dummies(data=df, columns=['A', 'B'])
   C  A_a  A_b  B_b  B_c
0  1  1.0  0.0  0.0  1.0
1  2  0.0  1.0  0.0  1.0
2  3  1.0  0.0  1.0  0.0