如何在python中没有任何ML框架的情况下实现一种热编码？

Question

没有任何MLframe的MNIST教程，但是陷入了一个热编码阶段

y包含数字图像的标签数据

0、1、2、3、4、5、6、7、8、9

其大小为（10000，）

我想将每个类别编号转换为一个热编码数组

0：1 0 0 0 0 0 0 0 0 0 01：0 1 0 0 0 0 0 0 0 0 02：0 0 1 0 0 0 0 0 0 0 0依此类推

所以我做了一个代码

import numpy as np
y_one=np.zeros(y.size, 10)
y_one[np.arange(y.size), y]=1

它说'不理解数据类型'

在这种情况下，如何在不使用sklearn或tf的情况下实现一种热编码？

Answer 1

将整数的元组传递为np.zeros的形状

y_one = np.zeros((y.size, 10))