Python科学计算（2）

Python科学计算（二）——正弦信号的时域波形与频谱图

Python科学计算（二）-- 正弦信号的时域波形与频域波形生成、计算与显示

# -*- coding: utf-8 -*-import numpy as npimport matplotlib.pyplot as plimport matplotlibimport mathimport randomrow = 4col = 4N = 500fs = 5n = [2*math.pi*fs*t/N for t in range(N)]axis_x = np.linspace(0,1,num=N)myfont = matplotlib.font_manager.FontProperties(fname='c:\\windows\\fonts\\fzshjw_0.ttf')#频率为5Hz的正弦信号x = [math.sin(i) for i in n]pl.subplot(221)pl.plot(axis_x,x)pl.title(u'5Hz的正弦信号', fontproperties=myfont)pl.axis('tight')#频率为5Hz、幅值为3的正弦+噪声x1 = [random.gauss(0,0.5) for i in range(N)]xx = []#有没有直接两个列表对应项相加的方式？？for i in range(len(x)):	xx.append(x[i]*3 + x1[i]) pl.subplot(222)pl.plot(axis_x,xx)pl.title(u'频率为5Hz、幅值为3的正弦+噪声', fontproperties=myfont)pl.axis('tight')#频谱绘制xf = np.fft.fft(x)xf_abs = np.fft.fftshift(abs(xf))axis_xf = np.linspace(-N/2,N/2-1,num=N)pl.subplot(223)pl.title(u'频率为5Hz的正弦频谱图', fontproperties=myfont)pl.plot(axis_xf,xf_abs)pl.axis('tight')#频谱绘制xf = np.fft.fft(xx)xf_abs = np.fft.fftshift(abs(xf))pl.subplot(224)pl.title(u'频率为5Hz的正弦频谱图', fontproperties=myfont)pl.plot(axis_xf,xf_abs)pl.axis('tight')

有疑问的留言讨论吧