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

Question

我怎么才能知道一个numpy数组或矩阵中最小值的（行，列）索引呢？

举个例子，如果我有一个数组 A = array([[1, 2], [3, 0]])，我想得到 (1, 1) 这个结果。

谢谢！

Answer 1

[已更正错别字]

还有一个简单的解决方案是

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

因为numpy.argmin返回的是按行优先顺序读取的索引

---

没错，你说得对，确实是个错别字，这个方法适用于方阵

Answer 2

使用 unravel_index 函数：

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