How to get N maximum values in a multi dimensional numpy array along a given axis(say 2)?

Question

Since argmax only gives one maximum values,how can we find atleast 2 or 3 elements instead of just one.

Currently my input is in the format np.argmax(array,axis=2) which is giving only one maximum and i have to extract 2 or 3 atleast from the array which is N-dimensional

What do you mean by "2 or 3 elements"? np.argmax extracts the argmax of all arrays in one dimension. Are you looking for the highest, second highest and third highest? — Martino
– Martino, Commented Jun 17, 2018 at 10:57
Question has nothing to do with machine-learning - kindly do not spam the tag (removed) — desertnaut
– desertnaut, Commented Jun 17, 2018 at 11:08
Possible duplicate of stackoverflow.com/questions/6910641/… — Merlin1896
– Merlin1896, Commented Jun 17, 2018 at 11:25

Carles · Accepted Answer · 2018-06-17 12:10:24Z

1

I would try to use the function called argpartition(). To get the indices of the two largest elements, do:

import numpy as np

a = np.array([9, 4, 4, 3, 3, 9, 0, 4, 6, 0])

ind = np.argpartition(a, -2)[-2:] 

ind
Out[13]: array([5, 0], dtype=int64)

a[ind]
Out[14]: array([9, 9])

answered Jun 17, 2018 at 12:10

Carles

2,8471 gold badge18 silver badges28 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

jpp · Accepted Answer · 2018-06-17 12:21:11Z

1

Using numpy.argsort. Data from @CarlesSansFuentes.

import numpy as np

a = np.array([9, 4, 4, 3, 3, 9, 0, 4, 6, 0])

args = np.argsort(-a)[:2]

array([0, 5], dtype=int64)

answered Jun 17, 2018 at 12:21

jpp

166k37 gold badges301 silver badges362 bronze badges

Collectives™ on Stack Overflow

How to get N maximum values in a multi dimensional numpy array along a given axis(say 2)?

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related