离散型字符特征处理

2020-02-29  本文已影响0人  Brian_mingzhi

离散特征的编码分为两种情况:

1、离散特征的取值之间没有大小的意义,比如color:[red,blue],此时可以使用one-hot编码;
实现:

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

2、离散特征的取值有大小的意义,比如size:[tall,grande,venti],此时可以使用数值的映射{tall:1,grande:2,venti:3}。

dic = {tall:1,grande:2,venti:3}
size = size.map(dic)
上一篇 下一篇

猜你喜欢

热点阅读