pd.get_dummy()函数使用

最新推荐文章于 2023-12-04 18:46:42 发布

我的心永远是冰冰哒

最新推荐文章于 2023-12-04 18:46:42 发布

阅读量732

点赞数 3

文章标签： pandas python 开发语言

本文链接：https://blog.csdn.net/qq_45759229/article/details/130035890

版权

import pandas as pd
df = pd.DataFrame([  
            ['green' , 'A'],   
            ['red'   , 'B'],   
            ['blue'  , 'A']])  
 
df.columns = ['color',  'class'] 
pd.get_dummies(df)

在这里插入图片描述
但是这里有个缺陷，就是返回的值全是bool型的，如果想把这个结果转成0和1

tmp = pd.DataFrame([  
        ['green' , True],   
        ['red'   , False],   
        ['blue'  , True]]) 
tmp.columns = ['color',  'class'] 
print(tmp)
pd.get_dummies(tmp,dtype=int)

在这里插入图片描述