how to find index of max value in numpy array code example

Example: numpy get max index

>>> a.argmax(axis=0)

array([1, 1, 0])