Running get_dummies on several DataFrame columns?

前端 未结 4 1883
爱一瞬间的悲伤
爱一瞬间的悲伤 2020-11-27 05:31

How can one idiomatically run a function like get_dummies, which expects a single column and returns several, on multiple DataFrame columns?

4条回答
  •  孤独总比滥情好
    2020-11-27 06:01

    With pandas 0.19, you can do that in a single line :

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

    Columns specifies where to do the 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
    

提交回复
热议问题