[Numpy-discussion] Find indices of largest elements

[Skipper Seabold]

On Wed, Apr 14, 2010 at 11:16 AM, Nikolaus Rath <Nikolaus@rath.org> wrote:
> Hello,
>
> How do I best find out the indices of the largest x elements in an
> array?
>
> Example:
>
> a = [ [1,8,2], [2,1,3] ]
> magic_function(a, 2) == [ (0,1), (1,2) ]
>
> Since the largest 2 elements are at positions (0,1) and (1,2).

Something like this might be made to work, if you want the max
elements along all the rows.

In [3]: a = np.asarray(a)

In [4]: a[range(len(a)),np.argmax(a, axis=1)]
Out[4]: array([8, 3])

Skipper