python - 在矩阵中查找最小值的索引

Question

我有一个 2-Dim 数组，其中包含给定拟合的残差平方和(此处不重要)。

RSS[i,j] = np.sum((spectrum_theo - sp_exp_int) ** 2)

我想找到具有最小值的矩阵元素及其在矩阵中的位置 (i,j)。找到最小元素就OK了:

RSS_min = RSS[RSS != 0].min()

但是对于索引，我试过:

ij_min = np.where(RSS == RSS_min)

这给了我:

ij_min =  (array([3]), array([20]))

我想获得:

ij_min = (3,20)

如果我尝试:

ij_min = RSS.argmin()

我得到:

ij_min = 0,

这是一个错误的结果。

在 Scipy 或其他地方是否存在可以做到这一点的函数？我在网上搜索过，但我找到的答案只包含 1-Dim 数组，而不是 2-Dim 或 N-Dim。

谢谢!

Answer 1

根据您现在所拥有的，最简单的解决方法就是从数组中提取元素作为最后一步:

# ij_min = (array([3]), array([20]))
ij_min = np.where(RSS == RSS_min)
ij_min = tuple([i.item() for i in ij_min])