NumPy: сортировка заданного сложного массива сначала по вещественной части, а затем по мнимой

Сортировка и поиск в NumPy: упражнение 6 с решением

Напишите программу NumPy для сортировки заданного сложного массива, сначала используя реальную часть, а затем мнимую.

Пример решения :

Код Python:

import numpy as np
complex_num = [1 + 2j, 3 - 1j, 3 - 2j, 4 - 3j, 3 + 5j]
print("Original array:")
print(complex_num)
print("\nSorted a given complex array using the real part first, then the imaginary part.")
print(np.sort_complex(complex_num))

Пример вывода:

 Исходный массив:
[(1 + 2j), (3-1j), (3-2j), (4-3j), (3 + 5j)]
Сортировал заданный комплексный массив сначала по вещественной части, а затем по мнимой.
[1. + 2.j 3.-2.j 3.-1.j 3. + 5.j 4.-3.j]

Редактор кода Python:

Есть другой способ решить это решение? Внесите свой код (и комментарии) через Disqus.

Предыдущий: Написать программу NumPy, чтобы получить индексы отсортированных элементов данного массива.
Далее: напишите программу NumPy для разбиения заданного массива в указанной позиции и переместите все меньшие значения элементов слева от раздела, а остальные значения вправо в произвольном порядке (на основе случайного выбора).