📜  一种热编码 numpy - Python 代码示例

📅  最后修改于: 2022-03-11 14:45:56.838000             🧑  作者: Mango

代码示例2
>>> 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.]])