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

Posted 2023-03-12

【中文标题】在二维矩阵中使用 numpy.nanargmin()

【英文标题】：Using numpy.nanargmin() in 2 dimensional matrix

【发布时间】：2017-03-17 14:53:35

【问题描述】：

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

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

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

还有……

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

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

【问题讨论】：

np.unravel_index(np.nanargmin(A), A.shape)（在1中你使用numpy.nanargmin(A)；为什么你在3中切换到A.nanargmin()？）

【参考方案1】：

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

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