numpy get index of max value code example

Example 1: python how to return max num index

import numpy as np

x=[1,2,3,4,5]

max_index=np.argmax(x) 
# return the index of the maximun element
# max_index has the value 4

Example 2: how to get index of max n values in an array

In [1]: import numpy as np

In [2]: arr = np.array([1, 3, 2, 4, 5])

In [3]: arr.argsort()[-3:][::-1]
Out[3]: array([4, 3, 1])