[Numpy-discussion] indexing to sort with argsort(..., axis=1)

Gökhan Sever gokhansever@gmail....
Mon Apr 12 22:01:48 CDT 2010


On Mon, Apr 12, 2010 at 9:41 PM, Angus McMorland <amcmorl@gmail.com> wrote:

> Hi all,
>
> I want to sort a 2d array along one dimension, with the indices returned by
> argsort, but the subsequent indexing syntax to get the sorted array is not
> obvious.
>
> The following works, but I wonder if there is a simpler way:
>
> a = np.random.random(size=(5,3))
> s = np.argsort(a, axis=1)
> sorted = a[:,s][np.eye(5,5, dtype=bool)] # it looks like this line could be
> simpler
>
> What's the correct, concise way to do this?
>

Why not just:

b = np.sort(a)

What advantage does argsort provides in this case?



-- 
Gökhan
-------------- next part --------------
An HTML attachment was scrubbed...
URL: http://mail.scipy.org/pipermail/numpy-discussion/attachments/20100412/f78e0a31/attachment.html 


More information about the NumPy-Discussion mailing list