argmax

argmax(values, **kws)[source]

Returns the indices of the maximum values along an axis.

Parameters:values (numpy.array | list) – Input array.
Returns:Indices of the maximum values
Return type:numpy.ndarray