确定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])