[Numpy-discussion] Find the N maximum values and corresponding indexes in an array

Keith Goodman kwgoodman at gmail.com
Wed Dec 2 21:48:14 EST 2009


On Wed, Dec 2, 2009 at 5:44 PM, Charles R Harris
<charlesr.harris at gmail.com> wrote:
>
>
> On Wed, Dec 2, 2009 at 6:32 PM, Anne Archibald <peridot.faceted at gmail.com>
> wrote:
>>
>> 2009/12/2 David Warde-Farley <dwf at cs.toronto.edu>:
>> > On 2-Dec-09, at 8:09 PM, Neal Becker wrote:
>> >
>> >> Not bad, although I wonder whether a partial sort could be faster.
>> >
>> > Probably (if the array is large) but depending on n, not if it's in
>> > Python. Ideal problem for Cython, though.
>>
>> How is Cython support for generic types these days? One wouldn't want
>> to have to write separate versions for each dtype...
>>
>
> It could be made part of the _sortmodule without much trouble, but if you
> are looking at the lower 30% I doubt it would do much more that buy you a
> factor of 2x. Would that matter?

That sounds fantastic!



More information about the NumPy-Discussion mailing list