美文网首页我爱编程
pandas的get_dummies

pandas的get_dummies

作者: dechuan | 来源:发表于2017-02-21 09:27 被阅读0次

    我理解get_dummies是将拥有不同值的变量转换为0/1数值。打个比方,小明有黄、红、蓝三种颜色的帽子,小明今天戴黄色帽子用1表示,红色帽子用2表示,蓝色帽子用3表示。但1、2、3数值大小本身是没有意义的,只是用于区分帽子的颜色,因此在实际分析时,需要将1、2、3转化为0、1,如下代码所示:

    import pandas as pd
    xiaoming=pd.DataFrame([1,2,3],index=['yellow','red','blue'],columns=['hat'])
    print(xiaoming)
    hat_ranks=pd.get_dummies(xiaoming['hat'],prefix='hat')
    print(hat_ranks.head())
    
            hat
    yellow    1
    red       2
    blue      3
            hat_1  hat_2  hat_3
    yellow      1      0      0
    red         0      1      0
    blue        0      0      1
    

    相关文章

      网友评论

        本文标题:pandas的get_dummies

        本文链接:https://www.haomeiwen.com/subject/qsuywttx.html