Answers for "numpy get index of n max values"

2

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
Posted by: Guest on June-07-2020
1

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])
Posted by: Guest on July-04-2020

Code answers related to "numpy get index of n max values"

Browse Popular Code Answers by Language