python - 确定Python NumPy中最高值(value)的索引

我想用每行最高最大值的索引生成一个数组。

a = np.array([ [1,2,3], [6,5,4], [0,1,0] ])
maxIndexArray = getMaxIndexOnEachRow(a)
print maxIndexArray

[[2], [0], [1]]

有一个np.argmax函数，但它似乎没有执行我想要的操作...

最佳答案

argmax()函数可以执行您想要的操作:

print a.argmax(axis=1)
array([2, 0, 1])

关于python - 确定Python NumPy中最高值(value)的索引，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/4150542/