仅作为记录,大佬请跳过。
转载大佬博主文章传送门
直接上代码
# -*- coding: utf-8 -*-
from scipy import signal
import numpy as np
import matplotlib.pyplot as pl
import matplotlib
import math
pl.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一(替换sans-serif字体)
pl.rcParams['axes.unicode_minus'] = False # 步骤二(解决坐标轴负数的负号显示问题)
ts=0.005
t=np.arange(0,1,ts)
fs=1/ts
axis_x = np.linspace(0, 1, num=len(t))
# 设置字体文件,否则不能显示中文
# myfont = matplotlib.font_manager.FontProperties(fname='c:\\windows\\fonts\\fzshjw_0.ttf')
# 频率为5Hz的正弦信号
x = np.sin(2*np.pi*5*t)
pl.subplot(221)
pl.plot(axis_x, x)
pl.title(u'5Hz的正弦信号')
pl.axis('tight')
xx = []
x1 = np.sin(2*np.pi*50*t)
for i in range(len(x)):
xx.append(x[i] + x1[i])
pl.subplot(222)
pl.plot(axis_x, xx)
pl.title(u'5Hz与50Hz的正弦叠加信号')
pl.axis('tight')
b, a = signal.butter(3, 0.08, 'low')
lf = signal.filtfilt(b, a, xx)
pl.subplot(223)
pl.plot(axis_x, lf)
pl.title(u'低通滤波后', )
pl.axis('tight')
b, a = signal.butter(3, 0.10, 'high')
sf = signal.filtfilt(b, a, xx)
pl.subplot(224)
pl.plot(axis_x, sf)
pl.title(u'高通滤波后')
pl.axis('tight')
# pl.show()
# 傅里叶
def fftspectrum(thesignal,fs): # add fs
nsel=thesignal.size
fsel=fs*(np.arange(0,nsel/2)/nsel) # add fs*
ysel=np.fft.fft(thesignal)
ysel=np.abs(ysel)
ysel=ysel[0:len(fsel)]
ysel=20*np.log(ysel)
# plt.figure()
pl.plot(fsel,ysel)
# plt.show()
# fs=200
pl.figure() # function needs
pl.subplot(221);fftspectrum(np.array(x),fs)
pl.subplot(222);fftspectrum(np.array(xx),fs)
pl.subplot(223);fftspectrum(np.array(lf),fs)
pl.subplot(224);fftspectrum(np.array(sf),fs)
pl.show() # function needs
结果展示: