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