在二维矩阵中使用 numpy.nanargmin()

Question

我正在尝试获取具有 nan 值的 numpy 二维数组 A 的 argmin。现在的问题是：

1. numpy.nanargmin(A) 只返回一个索引。

2. numpy.unravel_index(A.argmin(), A.shape) 返回[0,0]，因为它具有nan 值。

   还有……

3. numpy.unravel_index(A.nanargmin(), A.shape) 抛出错误：

   AttributeError Traceback（最近调用 最后）在（） ----> 1 np.unravel_index(dist.nanargmin(), dist.shape) AttributeError: 'numpy.ndarray' 对象没有属性 'nanargmin'

Answer 1

NumPy 数组只有argmin() 属性，但没有nanargmin() 属性。所以A.nanargmin() 不存在。

您可以改用numpy.argmin(A) 和numpy.nanargmin(A)。