要在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]