> Date: Wed, 22 Jun 2011 11:30:47 -0400 > From: Alex Flint <alex.fl...@gmail.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. Best, Radim _______________________________________________ NumPy-Discussion mailing list NumPy-Discussion@scipy.org http://mail.scipy.org/mailman/listinfo/numpy-discussion