NumPy 陣列中的虛數

1. 使用 numpy.complex 類將虛數儲存在 NumPy 陣列中
2. 使用 dtype 引數將虛數儲存在 numpy 陣列中

在 Python 中，我們可以使用實數和虛數。

為了表示一個複數，我們只需在末尾新增 j。例如，以下字串表示一個虛數。

a = 5 + 2j
print(a, type(a))

輸出：

(5+2j) <class 'complex'>

在本教程中，我們將學習如何處理 numpy 陣列中的虛數。

如前所述，我們可以簡單地通過新增 j 來建立複數。我們可以使用諸如 numpy.array()，numpy.arange() 之類的不同函式將該文字廣播到陣列中。有關 numpy 陣列中複數的各種示例，請參見下面的程式碼。

arr_1 = 1j * np.arange(5)
arr_2 = np.array([2 + 1j, 3 + 4j, 5 + 2j])
print(arr_1)
print(arr_2)

輸出：

[0.+0.j 0.+1.j 0.+2.j 0.+3.j 0.+4.j]
[2.+1.j 3.+4.j 5.+2.j]

使用 numpy.complex 類將虛數儲存在 NumPy 陣列中

建立複雜物件的另一種方法是使用 numpy 模組提供的 complex 類。這將返回一個複雜的物件，該物件可以再次儲存在陣列中，如先前方法所述。

例如，

a = np.complex(1 + 1j)
c = a * np.arange(5)
print(c)

輸出：

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

使用 dtype 引數將虛數儲存在 numpy 陣列中

在陣列中啟動虛數的另一種方法是通過在一些 numpy 陣列函式中指定 dtype 引數。眾所周知，我們可以使用 numpy.zeros() 和 numpy.ones() 函式分別建立 0 和 1 的陣列。在這裡，我們可以將 dtype 引數指定為 complex 以獲得具有複雜值的結果陣列。

以下程式碼對此進行了說明。

z = np.ones(4, dtype=complex) * 2
print(z)

輸出：

[2.+0.j 2.+0.j 2.+0.j 2.+0.j]