Determinar el índice de mayor valor en Python's NumPy

Question

Quiero generar una matriz con el índice del valor máximo más alto de cada fila.

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

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

Hay una función np.argmax pero no parece hacer lo que quiera ...

Answer 1

La función argmax() qué hacer lo que quiera:

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