NumPy 陣列中的虛數
Manav Narula
2023年1月30日
2021年4月29日
在 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]
Author: Manav Narula
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