帮我用rdann写一段提取MIT-BIH数据集中ECG标签的python代码

好的，以下是使用rdann提取MIT-BIH数据集中ECG标签的Python代码：

import os
import subprocess

WFDB = '/path/to/wfdb'  # wfdb库的路径
DATA = '/path/to/mitbih'  # MIT-BIH数据集的路径

def extract_labels(idx):
    rdann = os.path.join(WFDB, 'rdann')
    output = subprocess.check_output([rdann, '-r', idx, '-a', 'atr'], cwd=DATA)
    labels = (line.split() for line in output.strip().split("\n"))
    labels = [(l[0], int(l[1]), l[2], l[6] if len(l) == 7 else None) for l in labels]
    return labels

# 测试代码
labels = extract_labels('100')
print(labels)

这段代码会提取MIT-BIH数据集中 `100` 数据文件的ECG标签，并返回一个列表，列表中每个元素包含四个值，分别为：文件名、标签位置、标签类型和标签注释。