numpy.ndarray.argmax
#
method
ndarray.argmax(axis=None, out=None, *, keepdims=False)
Return indices of the maximum values along the given axis.
Refer to
numpy.argmax
for full documentation.
See also
equivalent function