Return indices of the maximum values along the given axis.
Refer to numpy.argmax for full documentation.
See also
Enter search terms or a module, class or function name.