pandas.get_dummies 的用法

get_dummies 是利用pandas实现one hot encode的方式。详细参数请查看官方文档
官方文档在这里

pandas.get_dummies(data, prefix=None, prefix_sep='_', dummy_na=False, columns=None, sparse=False, drop_first=False)[source]

例子:

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

df.columns = ['color',  'class'] 
pd.get_dummies(df) 

get_dummies 前:
get_dummies 前
get_dummies 后:
get_dummies 后

上述执行完以后再打印df 出来的还是get_dummies 前的图,因为你没有写

df = pd.get_dummies(df)

可以对指定列进行get_dummies

pd.get_dummies(df.color)

这里写图片描述

将指定列进行get_dummies 后合并到元数据中

df = df.join(pd.get_dummies(df.color))

这里写图片描述

猜你喜欢

转载自blog.csdn.net/maymay_/article/details/80198468