This is documentation for an old release of NumPy (version 1.18). Read this page Search for this page in the documentation of the latest stable release (version 2.2).

numpy.nanargmax

numpy.nanargmax(a, axis=None)

Return the indices of the maximum values in the specified axis ignoring NaNs. For all-NaN slices ValueError is raised. Warning: the results cannot be trusted if a slice contains only NaNs and -Infs.

Parameters

aarray_like

Input data.

axisint, optional

Axis along which to operate. By default flattened input is used.

Returns

index_arrayndarray

An array of indices or a single index value.

See also

argmax, nanargmin

Examples

>>>

>>> a = np.array([[np.nan, 4], [2, 3]])
>>> np.argmax(a)
0
>>> np.nanargmax(a)
1
>>> np.nanargmax(a, axis=0)
array([1, 0])
>>> np.nanargmax(a, axis=1)
array([1, 1])