[Numpy-discussion] argmax for top N elements
Wed Jun 22 14:08:25 CDT 2011
> Date: Wed, 22 Jun 2011 11:30:47 -0400
> From: Alex Flint <email@example.com>
> Subject: [Numpy-discussion] argmax for top N elements
> Is it possible to use argmax or something similar to find the locations of
> the largest N elements in a matrix?
I would also be interested in an O(N) argmax/argmin for indices of top k values in an array. I'm currently using argsort[:k] in a performance sensitive part and it's not ideal.
More information about the NumPy-Discussion