离散特征的编码分为两种情况:
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)
网友评论