Минимальный объем в формате (строка, столбец)

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 возвращает показание индекса в главном порядке строк

---

Да,Вы правы, это была опечатка, которая работала на квадратную матрицу