argmax

argmax(values, **kws)

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