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']))