pandas.get_dummies的用法（one hot encode）

get_dummies是利用pandas实现one hot encode编码的方式。

例子：

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

get_dummies()后：

pd.get_dummies(df)

也可以对指定的列进行编码：

pd.get_dummies(df['class'])

 

对指定的列进行编码后合并到原数据中：

df.join(pd.get_dummies(df['class']))