onehot encoder

sklearn

from sklearn.preprocessing import OneHotEncoder
labels = [[0,1,2,5]]
enc.fit(labels)
enc.transform(labels).toarray()
enc = OneHotEncoder()

>>> enc.transform(labels).toarray()
array([[1., 0., 0., 0.],
       [0., 1., 0., 0.],
       [0., 0., 1., 0.],
       [0., 0., 0., 1.]])

默认是按照数据的个数进行排序，如果不想的话可以自己写一个