numpy.fft.fft是如何工作的?

时间:2015-11-21 21:58:51

标签: python numpy signal-processing fft

我目前正试图从numpy中了解fft-function。为此,我测试了以下假设:
我有两个功能,f(x) = x^2g(x) = f'(x) = 2*x。根据傅里叶变换定律和wolfram alpha,它应该是G(w) = 2pi*i*F(w)(前因子可以变化,但应该只有一个常数因子)。在python中实现时,我写了

import numpy as np
def x2(x):
    return x*x
def nx(x):
    return 2*x

a = np.linspace(-3, 3, 16)
a1 = x2(a)
a2 = nx(a)

b1 = np.fft.fft(a1)
b2 = np.fft.fft(a2)

c = b1/b2

现在我期待c几乎不变的值,但我得到了

array([  1.02081592e+16+0.j        ,   1.32769987e-16-1.0054679j ,
         4.90653893e-17-0.48284271j,  -1.28214041e-16-0.29932115j,
        -1.21430643e-16-0.2j       ,   5.63664751e-16-0.13363573j,
        -5.92271642e-17-0.08284271j,  -4.21346622e-16-0.03978247j,
        -5.55111512e-16-0.j        ,  -5.04781597e-16+0.03978247j,
        -6.29288619e-17+0.08284271j,   8.39500693e-16+0.13363573j,
        -1.21430643e-16+0.2j       ,  -0.00000000e+00+0.29932115j,
        -0.00000000e+00+0.48284271j,   1.32769987e-16+1.0054679j ])

我的错误在哪里,我该如何按预期使用fft?

1 个答案:

答案 0 :(得分:5)

您提供的属性适用于Continuous Fourier transform(CFT)。 FFT计算的是Discrete Fourier transform(DFT),它与CFT有关但并不完全等效。

在某些条件下,DFT与CFT成比例是正确的:即在样本限制之外对函数进行充分采样(参见例如this book的附录E)。

上述建议的功能都不适用,因此DFT与CFT不成比例,您的数值结果反映了这一点。

这里有一些代码通过FFT确认您感兴趣的关系,使用适当采样的带限功能:

import numpy as np

def f(x):
    return np.exp(-x ** 2)
def fprime(x):
    return -2 * x * f(x)

a = np.linspace(-10, 10, 100)
a1 = f(a)
a2 = fprime(a)

b1 = np.fft.fft(a1)
b2 = np.fft.fft(a2)
omega = 2 * np.pi * np.fft.fftfreq(len(a), a[1] - a[0])

np.allclose(b1 * 1j * omega, b2)
# True