确定Python NumPy中最高值的索引

Question

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

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

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

有一个np.argmax函数，但它似乎没有做我想要的......

Answer 1

argmax()函数 执行您想要的操作：

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