将np-array从float转换为complex

Question

我有一个带有float数据类型的numpy-array，但是extern函数需要一个复杂的数据类型。当我使用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

是的，请使用astype()：

library