Определить индекс наивысшего значения в NumPy Python

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