60

如何知道 numpy 数组/矩阵的最小值的(行、列)索引?

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

谢谢!

4

2 回答 2

109

使用unravel_index

numpy.unravel_index(A.argmin(), A.shape)
于 2010-07-12T16:13:13.420 回答
3

[更正错字]

另一个简单的解决方案是

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

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


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

于 2018-03-27T08:25:08.480 回答