Python学习之：使用pandas产生one-hot编码

import pandas as pddata = [['red',3] ,['green',5] ,['yellow',4] ,['red',6]]columns = ['color','age']

df = pd.Dataframe(data,columns=columns)df

colorage0red31green52yellow43red6

# 通过 get_dummies 方法将 Series 变成 one-hot的形式one_hot_color = pd.get_dummies(df['color'] ,prefix='color' # 产生的 one-hot 的新列的前缀名称是 color ,prefix_sep='_') # 产生的 one-hot 的新列的分隔符是 _one_hot_color

color_greencolor_redcolor_yellow0010110020013010

# 通过 df.join 将数据合并df.join(one_hot_color)

coloragecolor_greencolor_redcolor_yellow0red30101green51002yellow40013red6010