Эффективно поиск индексов наибольших значений в массиве Numpy, не сортируя весь массив [дублировать] ⇐ Python

[Anonymous]

Я работаю с очень большими массивами Numpy (миллионы до миллиардов элементов), и мне нужно найти показатели N самых больших значений в массиве. Использование np.argsort () с последующим нарезом для получения последних элементов n слишком медленное и интенсивно для памяти для моего варианта использования, так как он сортирует весь массив. В идеале, решение будет иметь временную сложность лучше, чем O (n log n), и избегать создания больших промежуточных массивов. < /P>
Например, с учетом массива: < /p>
import numpy as np
arr = np.array([1, 5, 2, 8, 3, 9, 4, 7, 6])
N = 3
< /code>
Желаемый вывод будет индексом 3 самых больших значений (9, 8, 7): < /p>
[5, 3, 7] # Indices of 9, 8, and 7


Подробнее здесь: https://stackoverflow.com/questions/795 ... ay-without