高通滤波结果展示:
import random
import numpy as np
import pandas as pd
from scipy import signal
import matplotlib.pyplot as plt
def sine_generator(fs, sinefreq, duration):
T = duration
nsamples = fs * T
w = 2. * np.pi * sinefreq
t_sine = np.linspace(0, T, nsamples, endpoint=False)
y_sine = np.sin(w * t_sine) + random.random()
result = pd.DataFrame({
'data' : y_sine} ,index=t_sine)
return result
def sine_generator2(fs, sinefreq, duration):
T = duration
nsamples = fs * T
w = 2. * np.pi * sinefreq
t_sine = np.linspace(0, T, nsamples, endpoint=False)
y_sine = t_sine
result = pd.DataFrame({
'data' : y_sine} ,index=t_sine)
return result
def sine_generator3(fs, sinefreq, duration):
T = duration
nsamples = fs * T
w = 2. * np.pi * sinefreq
t_sine = np.linspace(0, T, nsamples, endpoint=False)
y_sine = t_sine * random.random()
result = pd.DataFrame({
'data' : y_sine} ,index=t_sine)
return result
def butter_highpass(cutoff, fs, order=5):
nyq = 0.5 * fs
normal_cutoff = cutoff / nyq
b, a = signal.butter(order, normal_cutoff, btype='high', analog=False)
return b, a
def butter_highpass_filter(data, cutoff, fs, order=5):
b, a = butter_highpass(cutoff, fs, order=order)
y = signal.filtfilt(b, a, data)
return y
fps = 30
sine_fq = 10 #Hz
duration = 10 #seconds
sine_5Hz = sine_generator(fps,sine_fq,duration)
sine_fq = 1 #Hz
duration = 10 #seconds
sine_1Hz = sine_generator(fps,sine_fq,duration)
t=sine_generator2(fps,sine_fq,duration)
k=sine_generator3(fps,sine_fq,duration)
sine = sine_5Hz + sine_1Hz + t + k
filtered_sine = butter_highpass_filter(sine.data,10,fps)
plt.rcParams['font.sans-serif']=['SimHei']
plt.rcParams['axes.unicode_minus'] = False
plt.figure(figsize=(20,10))
plt.subplot(211)
plt.plot(range(len(sine)),sine)
plt.title("原始信号")
plt.subplot(212)
plt.plot(range(len(filtered_sine)),filtered_sine)
plt.ylim(-2,2)
plt.title('高通滤波信号')
plt.show()
参考源:
https://stackoverflow.com/questions/39032325/python-high-pass-filter