将索引数组转换为1-hot编码的numpy数组

将索引数组转换为1-hot编码的numpy数组

您的数组

a

定义了输出数组中非零元素的列。您还需要定义行，然后使用花式索引：

>>> a = np.array([1, 0, 3])>>> b = np.zeros((a.size, a.max()+1))>>> b[np.arange(a.size),a] = 1>>> barray([[ 0.,  1.,  0.,  0.],       [ 1.,  0.,  0.,  0.],       [ 0.,  0.,  0.,  1.]])