Python 从二进制数据到整数

我有一个大小为

（N，）

的numpy向量，它包含

1-5

之间的整数，通过使用keras函数来构建相应的二进制矩阵，该矩阵的大小

（Nx5）

。例如，如果向量的第一个值是1，那么数组的第一行就是

（1，0，0，0，0）

。我怎么能做相反的事呢？通过使用一个值数组（也可以是双倍值）返回具有最高值的索引？是否有任何命令可以自动执行此操作

---

例如，我的输入可以是带有双精度的2D数组，例如一行可以是

[[0.025,0.022,0.58,0.011,0.22]…]

，并且该行的结果是

[3…]

如果

a

是您的数组，请使用：

a.argmax(axis=1)

这将返回数组每行的最大值索引

演示：

>>> a = np.array([[5,2,3],[1,3,1]])
>>> a.argmax(axis=1)
[0 1]

---

也许你可以发布一个输入示例并包含所需的输出？我的输入是一个二维数组，例如，一行可以是[[0.025,0.022,0.58,0.011,0.22]…]，结果是该行的[3…]。你肯定应该将该示例添加到你的问题中。您可以看看如何提供。