python带通滤波,python中的fft带通滤波器

What I try is to filter my data with fft. I have a noisy signal recorded with 500Hz as a 1d- array. My high-frequency should cut off with 20Hz and my low-frequency with 10Hz.

What I have tried is:

fft=scipy.fft(signal)

bp=fft[:]

for i in range(len(bp)):

if not 10

bp[i]=0

ibp=scipy.ifft(bp)

What I get now are complex numbers. So something must be wrong. What? How can I correct my code?

解决方案

It's worth noting that the magnitude of the units of your bp are not necessarily going to be in Hz, but are dependent on the sampling frequency of signal, you should use scipy.fftpack.fftfreq for the conversion. Also if your signal is real you should be using scipy.fftpack.rfft. Here is a minimal working example that filters out all frequencies less than a specified amount:

import numpy as np

from scipy.fftpack import rfft, irfft, fftfreq

time = np.linspace(0,10,2000)

signal = np.cos(5*np.pi*time) + np.cos(7*np.pi*time)

W = fftfreq(signal.size, d=time[1]-time[0])

f_signal = rfft(signal)

# If our original signal time was in seconds, this is now in Hz

cut_f_signal = f_signal.copy()

cut_f_signal[(W<6)] = 0

cut_signal = irfft(cut_f_signal)

We can plot the evolution of the signal in real and fourier space:

import pylab as plt

plt.subplot(221)

plt.plot(time,signal)

plt.subplot(222)

plt.plot(W,f_signal)

plt.xlim(0,10)

plt.subplot(223)

plt.plot(W,cut_f_signal)

plt.xlim(0,10)

plt.subplot(224)

plt.plot(time,cut_signal)

plt.show()