NumPy 陣列中的虛數

Manav Narula 2023年1月30日
  1. 使用 numpy.complex 類將虛數儲存在 NumPy 陣列中
  2. 使用 dtype 引數將虛數儲存在 numpy 陣列中
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]
作者: Manav Narula
Manav Narula avatar Manav Narula avatar

Manav is a IT Professional who has a lot of experience as a core developer in many live projects. He is an avid learner who enjoys learning new things and sharing his findings whenever possible.

LinkedIn