numpy.nanargmax()

numpy.nanargmax(a, axis=None) [source]

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:

Parameters:	a : array_like Input data. axis : int, optional Axis along which to operate. By default flattened input is used.
Returns:	index_array : ndarray An array of indices or a single index value.

a : array_like

Input data.

axis : int, optional

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

Returns:

index_array : ndarray

An array of indices or a single index value.

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])

Links:

https://docs.scipy.org/doc/numpy-1.11.0/reference/generated/numpy.nanargmax.html

doc_NumPy

2025-01-10 15:47:30

Comments