python - 如何从 NumPy 数组制作标签？

Question

我刚学Python，想请教一下..

比如我有

import numpy as np

a = np.array([[11, 12, 13],
              [14, 16, 13],
              [17, 15, 11],
              [12, 14, 15]])

我想找到该数组的标签

所以..第一行的最大值是13，那么标签结果= 3

第二行最大值为16，则标签结果=2

想要的结果是这样的

[3 2 1 3] or [[3]
              [2]
              [1]
              [3]]

Answer 1

你可以试试这个:

>>> import numpy as np

>>> a = np.array([[11, 12, 13],
              [14, 16, 13],
              [17, 15, 11],
              [12, 14, 15]])

>>> np.argmax(a, axis=1) + 1

array([3, 2, 1, 3], dtype=int64)

np.argmax给出指定轴上最大值的索引.所以，

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

然后您只需添加 1 即可。