NumPy proposes a way to get the index of the maximum value of an array via np.argmax.
np.argmax
I would like a similar thing, but returning the indexes of the
bottleneck has a partial sort function, if the expense of sorting the entire array just to get the N largest values is too great.
I know nothing about this module; I just googled numpy partial sort.
numpy partial sort