Return indices of the maximum values along the given axis.
Refer to numpy.argmax for full documentation.
See also