numpy.argmax — NumPy v2.3 Manual

numpy.argmax# numpy.argmax(a, axis=None, out=None, *, keepdims=<no value>)[source]# Returns the indices of the maximum values along an axis. Parameters: aarray_likeInput array. axisint, optionalBy default, the index is into the flattened array, otherwise along the specified axis. outarray, optionalIf provided, the result will be inserted into this array. It should be of the appropriate shape and d