Returns array of indices of the maximum values along the given axis. Masked values are treated as if they had the value fill_value.
Parameters : | axis : {None, integer}
fill_value : {var}, optional
out : {None, array}, optional
|
---|---|
Returns : | index_array : {integer_array} |
Examples
>>> a = np.arange(6).reshape(2,3)
>>> a.argmax()
5
>>> a.argmax(0)
array([1, 1, 1])
>>> a.argmax(1)
array([2, 2])