numpy.nanargmax#

numpy.nanargmax(a, axis=None, out=None, *, keepdims=<no value>)[源代码]#

返回指定轴上最大值的索引，忽略 NaN。对于全 NaN 切片，将引发 ValueError。警告：如果切片只包含 NaN 和 -Inf，结果可能不可靠。

参数:

a类数组对象: 输入数据。
axisint, optional: 操作的轴。默认情况下，使用展平的输入。
outarray, optional: 如果提供，结果将插入到此数组中。它应该具有适当的形状和数据类型。

New in version 1.22.0。
keepdimsbool，可选: 如果设置为 True，则被缩减的轴将作为大小为一的维度保留在结果中。使用此选项，结果将能正确地与数组进行广播。

New in version 1.22.0。

返回:

index_arrayndarray: 一个索引数组或单个索引值。

另请参阅

argmax, nanargmin

示例

>>> import numpy as np
>>> 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])