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

Keith Goodman kwgoodman@gmail....
Wed Dec 2 20:48:14 CST 2009


On Wed, Dec 2, 2009 at 5:44 PM, Charles R Harris
<charlesr.harris@gmail.com> wrote:
>
>
> On Wed, Dec 2, 2009 at 6:32 PM, Anne Archibald <peridot.faceted@gmail.com>
> wrote:
>>
>> 2009/12/2 David Warde-Farley <dwf@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