如何获得向量。从使用 argmax 的 2D numpy 数组？

debugcn 发表于 Dev

斯卡拉男孩

我有以下几点numpy.ndarray：

测试Arr：

array([[  2.55053788e-01,   6.25406146e-01,   1.19271643e-01,
          2.68359261e-04],
       [  2.59611636e-01,   0.19562805e-01,   1.20518960e-01,
          3.06535745e-01],
       [  8.52524495e-01,   5.24317825e-01,   1.22851081e-01,
          3.06610862e-04],
       [  2.55068243e-01,   6.24345124e-01,   1.20263465e-01,
          3.23178538e-04],
       [  2.46678621e-01,   6.29301071e-01,   1.23693809e-01,
          3.26490292e-04]], dtype=float32)

如果我这样做testVec = np.argmax(testArr)，我会得到一个号码。根据二维数组每行中的最大值，如何获得 0、1 或 2 的向量testArr？

预期输出：

[1, 3, 0, 1, 1]

珍娜112358

默认情况下， np.argmax 为您提供扁平数组中最大值的索引。要获得单个维度的最大值（例如，每行中的最大值），您必须指定关键字参数axis。这必须是一个整数：0 代表列，1 代表行。（如果您的数组有 n 维，则可以使用任何最多为 n-1 的整数。）

import numpy as np
testArr = np.array([[  2.55053788e-01,   6.25406146e-01,   1.19271643e-01,
          2.68359261e-04],
       [  2.59611636e-01,   0.19562805e-01,   1.20518960e-01,
          3.06535745e-01],
       [  8.52524495e-01,   5.24317825e-01,   1.22851081e-01,
          3.06610862e-04],
       [  2.55068243e-01,   6.24345124e-01,   1.20263465e-01,
          3.23178538e-04],
       [  2.46678621e-01,   6.29301071e-01,   1.23693809e-01,
          3.26490292e-04]], dtype=np.float32)
np.argmax(testArr, axis=1)
>>> array([1, 3, 0, 1, 1])

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。