On 30/10/2019 19:10, Neal Becker wrote: > max(axis=1)? Hi Neal,
I should have been more precise in stating the problem. Getting the values in the array for which I'm looking at the maxima is only one step in a more complex piece of code for which I need the indexes along the second axis of the array. I would like to avoid to have to iterate the array more than once. Thank you! Cheers, Dan > On Wed, Oct 30, 2019, 7:33 PM Daniele Nicolodi <dani...@grinta.net > <mailto:dani...@grinta.net>> wrote: > > 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 <mailto:NumPy-Discussion@python.org> > https://mail.python.org/mailman/listinfo/numpy-discussion > > > _______________________________________________ > NumPy-Discussion mailing list > NumPy-Discussion@python.org > https://mail.python.org/mailman/listinfo/numpy-discussion > _______________________________________________ NumPy-Discussion mailing list NumPy-Discussion@python.org https://mail.python.org/mailman/listinfo/numpy-discussion
