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

Как узнать индекс (строка, столбец) минимума массива/матрицы numpy?

Например, если A = array([[1, 2], [3, 0]]), я хочу получить (1, 1)

Спасибо!

Ответы

Ответ 1

Используйте unravel_index:

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

Ответ 2

[Исправлена опечатка]

Другое простое решение

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

Так как numpy.argmin возвращает чтение индекса в порядке следования строк

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