从浮点到复数的np数组转换

Question

我有一个带有float数据类型的numpy数组，但是一个外部函数需要一个复杂的数据类型。当我使用a.view(np.complex)时，实际值也用于复杂值，这会扰乱我的进一步计算，即：

a = [1, 2, 3]
b = a.view(np.complex)
> b = [1+1i, 2+2i, 3+3i]

有这样的命令吗

> b = [1+0i, 2+0i, 3+0i]

是吗？

Answer 1

是，使用^{}：

In [6]: b = a.astype(complex)
In [7]: b
Out[7]: array([ 1.+0.j,  2.+0.j,  3.+0.j])

In [8]: b.dtype
Out[8]: dtype('complex128')