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