NumPy 数组中的虚数

Manav Narula 2023年1月30日 2021年4月29日
  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]
Author: 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