How to get the index of a maximum element in a NumPy array along one axis

>>> a.argmax(axis=0)

array([1, 1, 0])

Leave a Comment