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])