要在NumPy中创建一个复数数组,可以使用numpy.array()函数并在输入值中包含虚部分。例如,可以使用以下代码创建一个包含复数的NumPy数组:
numpy.array()
import numpy as np arr = np.array([1 + 2j, 3 - 4j, 5 + 6j]) print(arr)
上面的代码将创建一个包含三个复数的NumPy数组,并输出如下:
[1.+2.j 3.-4.j 5.+6.j]