在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])