Re: sorting list of complex numbers

skip@xxxxxxxxx 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
.

Relevant Pages

  • sorting list of complex numbers
    ... but what if I want to sort by real, ... list with 20 elements where there are only 10 distinct reals but 20 distinct ... Schwartzian transform and map the complex numbers to tuples, sort that, then ...
    (comp.lang.python)
  • Re: Calculus XOR Probability
    ... intervals, or naturals, on the real line, as well as being the number of ... definition of 'the cardinality of S'. ... especially when it is not any definite sort of a number?" ... Well Ordering the Reals II: ...
    (sci.math)
  • Re: Calculus XOR Probability
    ... Tony Orlow wrote: ... *bounded* interval of reals can be said to have a length, ... I mean a number that denotes some number of some sort of units. ... No one can actually experience infinite quantities. ...
    (sci.math)
  • Re: Creating a "toy" OO/AO language...
    ... categorically specifies sin. ... this as a function defined over the Reals....if it is over the Reals, ... accuracy over some sort of floating point construct, ... Lahman...clearly using a finite expansion for the specification ...
    (comp.object)
  • Re: Dedekind Cuts, Fundamental Sequences: why?
    ... sets of naturals. ... view of what sort of reasoning about reals, functions of reals, and so on, ...
    (sci.math)
Loading