python使用陷波滤波器和带通滤波器对edf文件滤波

首先需要读取edf文件，可以使用pyedflib库：

import pyedflib

# 打开edf文件
f = pyedflib.EdfReader("example.edf")

# 获取信号通道数和采样点数
n_channels = f.signals_in_file
n_samples = f.getNSamples()[0]

# 读取信号
signals = np.zeros((n_channels, n_samples))
for i in range(n_channels):
    signals[i, :] = f.readSignal(i)

# 关闭文件
f.close()

接着可以使用scipy库中的signal模块进行陷波滤波：

from scipy import signal

# 设计陷波滤波器
f0 = 50  # 陷波中心频率 50Hz
Q = 10   # 品质因数
fs = 200 # 采样率
w0 = f0 / (fs / 2)
b, a = signal.iirnotch(w0, Q)

# 滤波
filtered_signals = signal.filtfilt(b, a, signals, axis=1)

对于带通滤波器，可以使用signal.iirfilter函数进行设计和滤波：

# 设计带通滤波器
f_low = 1  # 低截止频率 1Hz
f_high = 30 # 高截止频率 30Hz
order = 5   # 阶数

Wn = [f_low / (fs / 2), f_high / (fs / 2)]
b, a = signal.iirfilter(order, Wn, btype='band')

# 滤波
filtered_signals = signal.filtfilt(b, a, signals, axis=1)

最后可以使用pyedflib库将滤波后的信号写入edf文件：

import pyedflib

# 打开文件并创建输出文件
f_in = pyedflib.EdfReader('example.edf')
f_out = pyedflib.EdfWriter('example_filtered.edf', f_in)

# 将滤波后的信号写入输出文件
n_channels = f_in.signals_in_file
for i in range(n_channels):
    f_out.writePhysicalSamples(filtered_signals[i, :])

# 完成写入并关闭文件
f_out.close()
f_in.close()