离散特征的编码分为两种情况:
1、离散特征的取值之间没有大小的意义,比如color:[red,blue],那么就使用one-hot编码
2、离散特征的取值有大小的意义,比如size:[X,XL,XXL],那么就使用数值的映射{X:1,XL:2,XXL:3}
from sklearn import preprocessing
enc = preprocessing.OneHotEncoder()
enc.fit([[0,0,3],[1,1,0],[0,2,1],[1,0,2]]) #列表中代表四个样本
array = enc.transform([[0,1,3]]).toarray()
print(array)
结果:[[ 1. 0. 0. 1. 0. 0. 0. 0. 1.]]
[[0,0,3],[1,1,0],[0,2,1],[1,0,2]]
[0,0,3] #表示是该样本是,male Europe Internet Explorer
[1,1,0] #表示是 female,us Firefox
[0,2,1]#表示是 male Asia Chrome
[1,0,2] #表示是 female Europe Safari
摘自:https://blog.csdn.net/google19890102/article/details/44039761
具体参考:http://scikit-learn.org/stable/modules/preprocessing.html#encoding-categorical-features
此外也可以直接使用pandas 对数据进行onehot编码:
class_mapping中必须是set 否则就会对每一个属性编码,包括重复的属性。
如果直接使用get_dummies(df)则会为每一个唯一的属性值创建一列