numpy get index of max array code example

Example: numpy get max index

>>> a.argmax(axis=0)

array([1, 1, 0])