max(axis=1)? On Wed, Oct 30, 2019, 7:33 PM Daniele Nicolodi <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 > https://mail.python.org/mailman/listinfo/numpy-discussion >
_______________________________________________ NumPy-Discussion mailing list NumPy-Discussion@python.org https://mail.python.org/mailman/listinfo/numpy-discussion