jax.numpy.sort_complex

jax.numpy.sort_complex(a)[source]

Sort a complex array using the real part first, then the imaginary part.

LAX-backend implementation of sort_complex(). Original docstring below.

Parameters

a (array_like) – Input array

Returns

out – Always returns a sorted complex array.

Return type

complex ndarray

Examples

>>> np.sort_complex([5, 3, 6, 2, 1])
array([1.+0.j, 2.+0.j, 3.+0.j, 5.+0.j, 6.+0.j])
>>> np.sort_complex([1 + 2j, 2 - 1j, 3 - 2j, 3 - 3j, 3 + 5j])
array([1.+2.j,  2.-1.j,  3.-3.j,  3.-2.j,  3.+5.j])