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
一个索引数组或单个索引值。
示例
>>> 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])