I found the solution, we should use numpy's argsort(...):

How can we get the indices of values in the original list after sorting a
> list?
>
>>
>> for example:
>>
>> (Pdb) A=[ 1, -1, 0, 7, 9, 1.3, 2.9 ]
>> (Pdb) A.sort()
>> (Pdb) A
>> [-1, 0, 1, 1.3, 2.9, 7, 9]
>> (Pdb)
>>
>>
>> Now I want to have the original indices of the sorted list, i.e:
>>
>> [1, 2, 0, 5, 6, 3, 4]
>>
>>
(Pdb)  A=[ 1, -1, 0, 7, 9, 1.3, 2.9 ]
(Pdb) sort_index = numpy.argsort(A)
(Pdb) sort_index
array([1, 2, 0, 5, 6, 3, 4])
_______________________________________________
Tutor maillist  -  Tutor@python.org
To unsubscribe or change subscription options:
http://mail.python.org/mailman/listinfo/tutor

Reply via email to