Depends on how big your array is. Numpy C code is 150x+ faster than python
overhead. Fancy indexing can be expensive in my experience. Without trying
I'd guess arr[:, argmax(arr, axis=1)] does what you want, but even if it
is, try profiling the two and see. I highly doubt such would be even 1% o
Dear Python-Numpy community,
Few years ago I started to use a lot Python and Numpy for science. I'd like to
thanks all people who contribute to this fantastic community.
I used a lot Cython, Pythran and Numba and for the FluidDyn project, we created
Transonic, a pure Python package to easily ac