python实现傅立叶级数展开

相关文章：Python实现快速傅里叶变换（FFT）

傅立叶级数的介绍我就不说了，自己也是应用为主，之前一直觉得很难懂，但最近通过自己编程实现了一些函数的傅立叶级数展开之后对傅立叶 级数展开的概念比较清楚了

$$f(x)=\frac{a_0}{2} + \sum_{n=1}^{\infty}(a_n cos(nx)+b_n sin(nx))$$
$$a_n=\frac{1}{\pi}\int_{-\pi}^{\pi}f(x)cos(nx)dx$$
$$b_n=\frac{1}{\pi}\int_{-\pi}^{\pi}f(x)sin(nx)dx$$

(1)函数如下

$$\begin{equation} f(x)=\left\{ \begin{array}{lr} exp(x), &-\pi \leq x <0 \\ 1, & 0\leq x <\pi\\ \end{array} \right. \end{equation}$$
函数图象如下：

代码：

from pylab import *
x = mgrid[-10:10:0.02]  # 这里类似于MATLAB用冒号产生步长为0.02的序列，但是语法和MATLAB不同
n = arange(1,1000)
def fourier_transform():
    a0 = (1-exp(-pi))/pi+1
    s=a0/2
    for i in range(1,100,1):
        s0 = (  (1-(-1)**i*exp(-pi))/(pi*(1+i**2))*cos(i*x)+1/pi*( (-i*(1-(-1)**i*exp(-pi)))/(1+i**2) + (1-(-1)**i)/i ) * sin(i*x) )
        s=s+s0
    plot(x,s,'orange',linewidth=0.6)
    title('fourier_transform')
    show()    

fourier_transform()

结果如下：

（2）

$$\begin{equation} f(x)=\left\{ \begin{array}{lr} x+\pi, &-\pi \leq x <0 \\ \pi-x, & 0\leq x <\pi\\ \end{array} \right. \end{equation}$$

转换代码：

from pylab import *
def fourier1():
    s=pi/2

    for i in range(1,100,1):
        s0 = 2/pi*(1-(-1)**i)/i**2 * cos(i*x) 
        s=s+s0
    plot(x,s,'orange',linewidth=0.6)
    title('fourier1')
    show()
fourier1()

结果如下：

关于傅立叶级数展开的还有一些例子在我的另一篇博客http://blog.csdn.net/ouening/article/details/53199215 也有讨论，通过以上的例子可以很好直观感受傅立叶级数的作用，在电气工程上对分析流经变压器的谐波分量、整流装置输出谐波等分析具有重要作用。

还有一点就是python的强大功能，利用for循环语句可以方便地实现级数展开的求和运算，自己感觉编程起来比MATLAB/Octave跟快