Numpy最小（行，列）格式

Question

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

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

谢谢！

Answer 1

使用unravel_index：

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

Answer 2

[纠正错字]

另一个简单的解决方案是

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

由于numpy.argmin以行主要顺序

返回索引读数

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