How do I sort these?

Bengt Richter bokr at oz.net
Sat Oct 29 22:27:45 EDT 2005


On Fri, 28 Oct 2005 20:00:42 +0100, Steve Holden <steve at holdenweb.com> wrote:

>KraftDiner wrote:
>> I have two lists.
>> I want to sort by a value in the first list and have the second list
>> sorted as well... Any suggestions on how I should/could do this?
>> 
> >>> first = [1, 3, 5, 7, 9, 2, 4, 6, 8]
> >>> second = ['one', 'three', 'five', 'seven', 'nine', 'two', 'four', 
>'six', 'eight']
> >>> both = zip(first, second)
> >>> both.sort()
> >>> [b[0] for b in both]
>[1, 2, 3, 4, 5, 6, 7, 8, 9]
> >>> [b[1] for b in both]
>['one', 'two', 'three', 'four', 'five', 'six', 'seven', 'eight', 'nine']
> >>>
>
>You mean like this?
ISTM there could be a subtle requirement in the way the OP stated what he wanted to do.
I.e., it sounds like he would like to sort the first list and have a second list undergo
the same shuffling as was required to sort the first. That's different from having the
data of the second participate in the sort as order-determining data, if equals in the
first list are not to be re-ordered:

 >>> first = [2]*5 + [1]*5
 >>> first
 [2, 2, 2, 2, 2, 1, 1, 1, 1, 1]
 >>> sorted(first)
 [1, 1, 1, 1, 1, 2, 2, 2, 2, 2]
 >>> second = [chr(ord('A')+i) for i in xrange(9,-1,-1)]
 >>> second
 ['J', 'I', 'H', 'G', 'F', 'E', 'D', 'C', 'B', 'A']
 >>> sorted(second)
 ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J']

Now the zipped sort unzipped:
 >>> zip(*sorted(zip(first,second)))
 [(1, 1, 1, 1, 1, 2, 2, 2, 2, 2), ('A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J')]

Now suppose we sort the first and use the elements' indices to preserve order where equal
 >>> sorted((f,i) for i,f in enumerate(first))
 [(1, 5), (1, 6), (1, 7), (1, 8), (1, 9), (2, 0), (2, 1), (2, 2), (2, 3), (2, 4)]

Separate out the first list elements:
 >>> [t[0] for t in sorted((f,i) for i,f in enumerate(first))]
 [1, 1, 1, 1, 1, 2, 2, 2, 2, 2]

Now select from the second list, by first-element position correspondence:
 >>> [second[t[1]] for t in sorted((f,i) for i,f in enumerate(first))]
 ['E', 'D', 'C', 'B', 'A', 'J', 'I', 'H', 'G', 'F']

Which did the OP really want? ;-)

Regards,
Bengt Richter



More information about the Python-list mailing list