数值计算-利用Python实现快速傅里叶变换(FFT)(不调用函数库)
快速傅里叶变换介绍
首先介绍离散傅里叶变换(DFT),不管是正变换还是逆变换,计算公式可以归结为
X(k)=∑n=0N−1x(n)WNnk X(k)=\sum_{n=0}^{N-1}x(n)W_N^{nk} X(k)=n=0∑N−1x(n)WNnk
其中,NNN为抽样点的个数(一般为2n2^n2n), x(n)x(n)x(n)为已知数据(拟合函数时为节点函数值), WN=e−i2πnW_N=e^{-i\frac{2\pi}{n}}WN=e−in2π为旋转因子。
而快速傅里叶变换(FFT),只是对DFT在算法上进行一定的幼化,降低复杂度,同样可以总结为一个公式:
X(k)={
∑n=0N/2−1x1(r)WN/2rk+WNk∑n=0N/2−1x2(r)WN/2rkk=0,1,⋯ ,N2−1∑n=0N/2−1x1(r)WN/2rk′+WNk′∑n=0N/2−1x2(r)WN/2rk′k=k′+N2=N2,N2+1,⋯ ,NX(k)=\left\{ \begin{aligned} &\sum_{n=0}^{N/2-1}x_1(r)W_{N/2}^{rk}+W_N^{k}\sum_{n=0}^{N/2-1}x_2(r)W_{N/2}^{rk} \quad k=0,1,\cdots,\frac{N}{2}-1\\ &\sum_{n=0}^{N/2-1}x_1(r)W_{N/2}^{rk'}+W_N^{k'}\sum_{n=0}^{N/2-1}x_2(r)W_{N/2}^{rk'} \quad k=k'+\frac{N}{2}=\frac{N}{2},\frac{N}{2}+1,\cdots,N \end{aligned} \right. X(k)=⎩⎪⎪⎪⎪⎪⎨⎪⎪⎪⎪⎪⎧