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