Как узнать индекс (строка, столбец) минимума массива/матрицы 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 возвращает чтение индекса в порядке следования строк
Да, вы правы, это была опечатка, которая работала на квадратную матрицу