如何将复数放入numpy的数组中？

Question

>>> import numpy as np
>>> A = np.zeros((3,3))
>>> A[0,0] = 9
>>> A
array([[ 9.,  0.,  0.],
       [ 0.,  0.,  0.],
       [ 0.,  0.,  0.]])
>>> A[0,1] = 1+2j
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: can't convert complex to float
>>> A[0,1] = np.complex(1, 2)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: can't convert complex to float

根据我的示例代码，我试图将复数放入numpy的数组中，但它没有起作用。也许我错过了一些基本的东西。

- fronthem

Answer 1

如果你想创建一个包含复杂值的数组，需要向numpy指定一个复杂类型：

>>> A = np.zeros((3,3), dtype=complex)

>>> print A
[[ 0.+0.j  0.+0.j  0.+0.j]
 [ 0.+0.j  0.+0.j  0.+0.j]
 [ 0.+0.j  0.+0.j  0.+0.j]]

>>> A[0,0] = 1. + 2.j

>>> print A
[[ 1.+2.j  0.+0.j  0.+0.j]
 [ 0.+0.j  0.+0.j  0.+0.j]
 [ 0.+0.j  0.+0.j  0.+0.j]]

- talonmies

Answer 2

你需要转换数组的类型。

示例：

>>> from numpy import array, dtype
>>> A = array([1, 2, 3])
>>> A[0] = 1+2j
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: can't convert complex to long
>>> B = A.astype(dtype("complex128"))
>>> B[0] = 1+2j
>>> B
array([ 1.+2.j,  2.+0.j,  3.+0.j])
>>> 

- James Mills

Answer 3

我没有足够的声望评论talonmies's answer（并且建议的编辑队列已满），所以我必须在这里发表。

np.complex将被弃用，因此您需要设置dtype=complex

以下是他们版本1.20.0发布说明中的弃用列表：https://numpy.org/doc/stable/release/1.20.0-notes.html?highlight=complex#deprecations

- lumalot

---

这并没有回答问题。一旦您拥有足够的声望，您将能够评论任何帖子；相反，提供不需要询问者澄清的答案。- 来自审核 - Sylvester Kruin

你的回答可以通过提供更多支持信息来改进。请编辑以添加进一步的细节，例如引用或文档，以便他人可以确认你的答案是正确的。您可以在帮助中心中找到有关如何编写良好答案的更多信息。 - Community