[Numpy-discussion] argmax() indexes to value

Daniele Nicolodi daniele at grinta.net
Wed Oct 30 19:23:05 EDT 2019


this is a very basic question, but I cannot find a satisfying answer.
Assume a is a 2D array and that I get the index of the maximum value
along the second dimension:

i = a.argmax(axis=1)

Is there a better way to get the value of the maximum array entries
along the second axis other than:

v = a[np.arange(len(a)), i]


Thank you.


More information about the NumPy-Discussion mailing list