sorting list of complex numbers
Diez B. Roggisch
deets at nospam.web.de
Sun Nov 9 10:19:50 EST 2008
skip at pobox.com schrieb:
> The thread on sorting in Python 3 got me to thinking. How could I sort a
> list of complex numbers using key?
>
> >>> lst = [random.random()+random.random()*1j for i in range(10)]
> >>> lst
> [(0.32672251849959244+0.41428983433288791j), (0.35238056484609881+0.92758203977208264j), (0.19337824038125528+0.16527285180541951j), (0.47569307114525849+0.72381960418815128j), (0.21498813135082351+0.2046308266222292j), (0.30142745756937939+0.35216751451102601j), (0.77355676386939132+0.0023447924287695043j), (0.2547736124606309+0.52234837788936905j), (0.38349189081350132+0.62017617694427096j), (0.58362096773561245+0.87702443273108477j)]
>
> As expected:
>
> >>> sorted(lst)
> Traceback (most recent call last):
> File "<stdin>", line 1, in <module>
> TypeError: no ordering relation is defined for complex numbers
>
> This works:
>
> >>> sorted(lst, key=lambda x: x.real)
> [(0.19337824038125528+0.16527285180541951j), (0.21498813135082351+0.2046308266222292j), (0.2547736124606309+0.52234837788936905j), (0.30142745756937939+0.35216751451102601j), (0.32672251849959244+0.41428983433288791j), (0.35238056484609881+0.92758203977208264j), (0.38349189081350132+0.62017617694427096j), (0.47569307114525849+0.72381960418815128j), (0.58362096773561245+0.87702443273108477j), (0.77355676386939132+0.0023447924287695043j)]
>
> but what if I want to sort by real, then imaginary parts? Here's a longer
> list with 20 elements where there are only 10 distinct reals but 20 distinct
> imaginaries:
>
> >>> pprint.pprint(lst)
> [(1+2.73j),
> (9+3.77j),
> (7+27j),
> (8+28j),
> (2+2.8600000000000003j),
> (4+3.1200000000000001j),
> (2+22j),
> (9+29j),
> (3+2.9900000000000002j),
> (6+26j),
> 2.6000000000000001j,
> (8+3.6400000000000001j),
> (3+23j),
> (5+3.25j),
> (1+21j),
> (5+25j),
> 20j,
> (6+3.3799999999999999j),
> (7+3.5100000000000002j),
> (4+24j)]
>
> I can sort by the real parts just fine:
>
> >>> lst.sort(key=lambda x: x.real)
> >>> pprint.pprint(lst)
> [2.6000000000000001j,
> 20j,
> (1+2.73j),
> (1+21j),
> (2+2.8600000000000003j),
> (2+22j),
> (3+2.9900000000000002j),
> (3+23j),
> (4+3.1200000000000001j),
> (4+24j),
> (5+3.25j),
> (5+25j),
> (6+26j),
> (6+3.3799999999999999j),
> (7+27j),
> (7+3.5100000000000002j),
> (8+28j),
> (8+3.6400000000000001j),
> (9+3.77j),
> (9+29j)]
>
> but how do I then do a secondary sort by the imaginary part, preserving the
> existing ordering on the real parts? Seems like I have to resort to a
> Schwartzian transform and map the complex numbers to tuples, sort that, then
> map them back. With the cmp key it would have been a fairly trivial task to
> define the desired compare-real-then-imag function.
>
> Is there a way to do this using just the key arg, no extra data structures?
Doesn't (untested)
key=lambda x: (x.real, x.imag)
work?
Diez
More information about the Python-list
mailing list