Hello,

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.

Cheers,
Daniele
_______________________________________________
NumPy-Discussion mailing list
NumPy-Discussion@python.org
https://mail.python.org/mailman/listinfo/numpy-discussion

Reply via email to