python - （行，列）格式的 Numpy 最小值

Question

如何知道 numpy 数组/矩阵的最小值的（行、列）索引？

例如，如果A = array([[1, 2], [3, 0]])，我想得到(1, 1)

谢谢！

score 109 · Accepted Answer

109

numpy.unravel_index(A.argmin(), A.shape)

于 2010-07-12T16:13:13.420 回答

score 3 · Accepted Answer

[更正错字]

另一个简单的解决方案是

ri, ci = A.argmin()//A.shape[1], A.argmin()%A.shape[1]

由于 numpy.argmin 以行优先顺序返回索引读数

是的，你是对的，这是一个错字，适用于方阵

2 回答 2