If storing the array as a list of numbers isn't problematic, you can use
import heapqheapq.nlargest(N, a)to get the N largest members.
↧
Answer by Mike Graham for A fast way to find the largest N elements in an numpy array
↧