给定一维索引数组:

a = array([1, 0, 3])

我想将它一次性编码为二维数组:

b = array([[0,1,0,0], [1,0,0,0], [0,0,0,1]])

最佳答案

创建一个具有足够列的归零数组 b，即 a.max() + 1。
然后，对于每一行 i，将第 a[i] 列设置为 1。

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

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

https://stackoverflow.com/questions/29831489/