心电图testbench

心电图testbench是用于测试心电图(ECG)信号处理算法的工具。它可以模拟ECG信号，并提供一系列测试用例来验证算法的准确性和性能。

以下是一个心电图testbench的示例代码：

import numpy as np
import matplotlib.pyplot as plt

# 生成模拟的ECG信号
def generate_ecg_signal(duration, sampling_rate):
    t = np.arange(0, duration, 1/sampling_rate)
    ecg_signal = np.sin(2*np.pi*1*t) + 0.5*np.sin(2*np.pi*2*t) + 0.2*np.sin(2*np.pi*3*t)
    return ecg_signal

# 模拟心电图testbench
def ecg_testbench(ecg_signal, sampling_rate):
    # 在这里可以调用你的心电图处理算法进行峰值检测和心率计算
    # 这里只是一个示例，使用简单的阈值法进行峰值检测和心率计算
    threshold = 0.5
    peaks = []
    for i in range(1, len(ecg_signal)-1):
        if ecg_signal[i] > threshold and ecg_signal[i] > ecg_signal[i-1] and ecg_signal[i] > ecg_signal[i+1]:
            peaks.append(i)
    
    heart_rate = len(peaks) / (len(ecg_signal) / sampling_rate) * 60
    
    # 绘制心电图和峰值点
    plt.figure()
    plt.plot(ecg_signal)
    plt.scatter(peaks, [ecg_signal[i] for i in peaks], color='red')
    plt.xlabel('Time')
    plt.ylabel('Amplitude')
    plt.title('ECG Signal')
    plt.show()
    
    return peaks, heart_rate

# 设置参数
duration = 10  # 信号时长为10秒
sampling_rate = 1000  # 采样率为1000Hz

# 生成模拟的ECG信号
ecg_signal = generate_ecg_signal(duration, sampling_rate)

# 运行心电图testbench
peaks, heart_rate = ecg_testbench(ecg_signal, sampling_rate)

print("Detected peaks:", peaks)
print("Heart rate:", heart_rate, "bpm")

这段代码首先生成了一个模拟的ECG信号，然后调用心电图testbench函数进行峰值检测和心率计算。在这个示例中，使用了简单的阈值法进行峰值检测，并根据检测到的峰值点计算心率。最后，绘制了ECG信号和检测到的峰值点。