16
Як отримати індекси N максимальних значень у масиві NumPy?
NumPy пропонує спосіб отримати індекс максимального значення масиву через np.argmax. Мені б хотілося подібного, але повертаючи індекси Nмаксимальних значень. Наприклад, якщо у мене є масив, [1, 3, 2, 4, 5], function(array, n=3)буде повертати індекси , [4, 3, 1]які відповідають елементам [5, 4, 3].
482
python
numpy
max
numpy-ndarray