我试图再次翻译一些matlab代码,我遇到了另一个pickle。代码本身非常简单,它只是一个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代码,请告诉我。