dask.array.nanargmax

dask.array.nanargmax

dask.array.nanargmax(a, axis=None, keepdims=False, split_every=None, out=None)[源代码]

返回指定轴上最大值的索引,忽略 NaN。对于全为 NaN 的切片,会引发 ValueError。警告:如果切片仅包含 NaN 和 -Infs,结果将不可信。

此文档字符串是从 numpy.nanargmax 复制的。

Dask 版本可能存在一些不一致性。

参数
aarray_like

输入数据。

int, 可选

操作所沿的轴。默认使用展平的输入。

数组,可选

如果提供,结果将被插入到这个数组中。它应该具有适当的形状和数据类型。

1.22.0 新版功能.

keepdimsbool, 可选

如果设置为 True,则被缩减的轴将作为大小为 1 的维度保留在结果中。通过此选项,结果将正确地与数组进行广播。

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