numpy.sort_complex

numpy.sort_complex(a)

Функция sort_complex() возвращает отсортированную по последней оси копию массива комплексных чисел.

Параметры:
a - массив NumPy или подобный массиву объект.
Исходный массив.
Возвращает:
ndarray - массив NumPy
отсортированную по первой оси копию исходного массива a с той же формой и типом данных.
Смотрите так же:
sort


Примеры

>>> import numpy as np
>>> 
>>> a_r = np.random.randint(0, 3, 10)
>>> a_i = np.random.randint(0, 20, 10)
>>> a = a_r + a_i*1j
>>> a
array([1.+10.j, 2.+16.j, 2.+17.j, 2.+14.j, 1.+11.j, 2. +5.j, 0.+12.j,
       0. +2.j, 1. +8.j, 2.+13.j])
>>> 
>>> np.sort_complex(a)
array([0. +2.j, 0.+12.j, 1. +8.j, 1.+10.j, 1.+11.j, 2. +5.j, 2.+13.j,
       2.+14.j, 2.+16.j, 2.+17.j])

Многомерные массивы сортируются только по последней оси:

>>> b_r = np.random.randint(0, 3, (3, 4))
>>> b_i = np.random.randint(0, 20, (3, 4))
>>> b = b_r + b_i*1j
>>> b
array([[0.+13.j, 1. +8.j, 2.+13.j, 2. +3.j],
       [0. +4.j, 0. +5.j, 0.+11.j, 0. +0.j],
       [2.+13.j, 1. +2.j, 0. +1.j, 1. +3.j]])
>>> 
>>> np.sort_complex(b)
array([[0.+13.j, 1. +8.j, 2. +3.j, 2.+13.j],
       [0. +0.j, 0. +4.j, 0. +5.j, 0.+11.j],
       [0. +1.j, 1. +2.j, 1. +3.j, 2.+13.j]])