【pandas】get_dummies将类别特征转为onehot

今天又学了一招，pandas将str类型的类别特征，转为one-hot。注意一定要是str类型，int类型报错。

pandas.get_dummies 的用法_数据分析-CSDN博客

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 后：每个类别自成一列。

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

df = pd.get_dummies(df)

可对指定列进行get_dummies，指定color列one-hot

pd.get_dummies(df.color)

 

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

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

 

 蓝色的，原color特征列要删掉。 

不知道如果预测的时候，怎么搞？？需要第一次将one-hot数据保存下来，然后redis里输入color直接调用one-hot结果？？？