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函数，但它似乎不符合我的需求...

- Jonathan

Answer 1

argmax()函数确实可以达到你想要的效果：

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

- Sven Marnach

---

这也存在于顶级函数中，因此以下内容也适用：maxvals = np.argmax(a, axis=1)。 - Robin