基于MIT数据集ECG信号的QRS波定位并绘制出轨迹的Python代码

以下是基于MIT数据集ECG信号的QRS波定位并绘制出轨迹的Python代码：

import wfdb
import matplotlib.pyplot as plt
import numpy as np

# 读取数据
record = wfdb.rdrecord('mitdb/100', sampto=10000)
annotation = wfdb.rdann('mitdb/100', 'atr', sampto=10000)

# 获取QRS波
qrs_inds = annotation.sample
qrs_vals = record.p_signal[qrs_inds]

# 绘制ECG信号
plt.figure(figsize=(10,5))
plt.plot(record.p_signal)
plt.xlabel('Sample')
plt.ylabel('Amplitude')
plt.title('ECG Signal')
plt.show()

# 绘制QRS波轨迹
plt.figure(figsize=(10,5))
plt.plot(record.p_signal)
plt.plot(qrs_inds, qrs_vals, 'ro')
plt.xlabel('Sample')
plt.ylabel('Amplitude')
plt.title('QRS Trajectory')
plt.show()

解释：

1. 使用 `wfdb.rdrecord()` 读取数据，其中 `sampto` 参数指定读取的采样点数；
2. 使用 `wfdb.rdann()` 读取注释信息，其中 `sampto` 参数同样指定读取的采样点数；
3. 从注释信息中获取QRS波的位置和值；
4. 使用 `plt.plot()` 绘制ECG信号和QRS波轨迹，其中 `ro` 参数代表红色圆点。