numpy array find the index of the max value code example

Example: numpy get max index

>>> a.argmax(axis=0)

array([1, 1, 0])