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