Numpy.sort_complex()函数

Numpy中的sort_complex()函数对复数类型的数组进行排序，它使用复数的实部作为第一关键字，虚部作为第二关键字。

语法

numpy.sort_complex(array[, axis=-1])

参数：

• array: 输入数组，类型为复数。
• axis：要沿着其排序的轴（如果存在）。默认为-1，即沿最后一个轴排序。

返回值：

• 排序后的数组。

示例

import numpy as np

arr = np.array([2+1j, 1-2j, 3+4j, 2-3j, 3+1j])
print(np.sort_complex(arr))

输出结果：

[1.-2.j 2.-3.j 2.+1.j 3.+1.j 3.+4.j]

该函数默认沿着最后一个轴排序。因此，上面的示例中，每个复数的实部和虚部都会根据其大小进行排序。

如果我们在沿着第一个轴排序，则输出结果如下：

print(np.sort_complex(arr, axis=0))

输出结果：

[2.-3.j 1.-2.j 2.+1.j 3.+1.j 3.+4.j]

在这个示例中，第一列中的每个复数的实部和虚部都会根据其大小进行排序。

注意点

• 该函数同时支持一维和多维数组。
• 如果数组中包含nan，则该函数会将nan的位置放在最后。
• 该函数在排序时不会修改原始数组，而是返回排序后的副本。

以上就是numpy.sort_complex()函数的介绍，希望能帮助大家处理复数类型的排序问题。