热编码

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]

df = pd.get_dummies(df)
可以对指定列进行get_dummies
pd.get_dummies(df.color)