numpy之one-hot

a = np.array([1, 0, 3])#数据列，one-hot编码之后非零的列
n_class = a.max() + 1
n_sample = a.shape[0]
b = np.zeros((n_sample, n_class))#3个样本，4个类别
b[:, a] = 1#非零列赋值为1
b

array([[ 0.,  1.,  0.,  0.],
[ 1.,  0.,  0.,  0.],
[ 0.,  0.,  0.,  1.]])
>>>

values = np.array([1, 0, 3])
n_values = np.max(values) + 1
np.eye(n_values)[values]

array([[ 0.,  1.,  0.,  0.],
[ 1.,  0.,  0.,  0.],
[ 0.,  0.,  0.,  1.]])

np.eye(4,3)[np.array([1, 1, 1, 1])]

array([[ 0.,  1.,  0.],
[ 0.,  1.,  0.],
[ 0.,  1.,  0.],
[ 0.,  1.,  0.]])

values = np.array([1, 0, 3,1,2,1,2])
n_values = np.max(values) + 1
np.eye(n_values)[values]

array([[ 0.,  1.,  0.,  0.],
[ 1.,  0.,  0.,  0.],
[ 0.,  0.,  0.,  1.],
[ 0.,  1.,  0.,  0.],
[ 0.,  0.,  1.,  0.],
[ 0.,  1.,  0.,  0.],
[ 0.,  0.,  1.,  0.]])