Python Numpy Определить индекс массива, который содержит общее значение max / min

Question

Допустим, у меня есть Numpy -й массив, как показано ниже.

arr = np.array([[2, 3, 9], [1,6,7], [4, 5, 8]])

Массив numpy представляет собой матрицу 3 x 3.

Что я хочу сделать, так это найти Индекс строки, который содержит общее максимальное значение.

Например, в приведенном выше примере максимальное значение для всего ndarray равно 9. Поскольку он расположен в первой строке, я хочу вернуть индекс 0.

Какой самый оптимальный способ сделать это?

Я знаю, что

np.argmax(arr)

Может возвращать индекс максимума для всего массива, для каждой строки или для каждого колонка. Однако существует ли метод, который может возвращать индекс строки, которая содержит общее максимальное значение?

Было бы также хорошо, если бы это также можно было легко изменить на столбец или найти минимум, et c.

Я не уверен, как go об этом без использования al oop и нахождения общего максимума при сохранении переменной для индекса, однако я чувствую, что это неэффективно для numpy и что должен быть лучшим методом.

Любая помощь или мысли приветствуются.

Спасибо за чтение.

Answer 1

Используйте , где с amax , amin .

import numpy as np

arr = np.array([[2, 3, 9], [1,6,7], [4, 5, 8]])

max_value_index = np.where(arr == np.amax(arr))
min_value_index = np.where(arr == np.amin(arr))

Выход:

(array([0]), array([2]))
(array([1]), array([0]))