numpy的复杂数字问题

Question

我正在尝试再次翻译一些matlab代码而且我遇到了另一个问题。代码本身非常简单，它只是一个4节点旋转因子的演示。这是我的尝试：

from numpy import *
from matplotlib import pyplot as plt 


x = zeros(4)
x[-1+1] = 0
x[0+1] = 1
x[1+1] = 1
x[2+1] = 0

z = 0 - 1j
W4 = exp(z*2*pi/4)
W0 = W4 ** 0
W1 = W4 ** 1
W2 = W4 ** 2
W3 = W4 ** 3


X = zeros(4)
X[-1+1] = (x[-1+1] + x[1+1]*W0) + W0*(x[0+1] + x[2+1]*W0)
X[0+1] = (x[-1+1] + x[1+1]*W2) + W1*(x[0+1] + x[2+1]*W2)
X[1+1] = (x[-1+1] + x[1+1]*W0) + W2*(x[0+1] + x[2+1]*W0)
X[2+1] = (x[-1+1] + x[1+1]*W2) + W3*(x[0+1] + x[2+1]*W2)


fx = fft.fft(x) 

plt.plot(X)
plt.plot(fx, 'ro')
plt.title("Results 4-point hand programmed FFT (blue) and the PYTHON routine (red o)")
plt.show()

以下是输出图像。第一个是用（几乎）相同的matlab代码运行的，第二个是上面python代码的图像。

对于第24到27行，它给出了错误“ComplexWarning：将复杂值转换为实际丢弃虚部”。现在我不习惯在python中处理复杂的数字。我尝试在所有变量中添加一个复杂的组件，但它给了我一个与matlab相关的图表。思考？如果您希望我发布matlab代码，请告诉我。

Answer 1

当您指定数组x和X时，您需要确保它具有复杂的数据类型，即：

x = zeros((4),dtype=complex)

编辑：

要修复绘图，您需要绘制实部和虚部：

plt.plot(X.real,X.imag)
plt.plot(fx.real,fx.imag, 'ro')

这给了我：

....看起来像你的Matlab图。