将tele数据集写为wfdb可用的数据集
TELE_data = np.load('Tele.npy', allow_pickle=True)
for i in range(250):
dat = TELE_data[i]
# print(dat['labels'])
sample = np.where(dat['qrs']==1)[0]
symbol = np.array(['N']*len(sample))
signals = []
signals.append(dat['signal'].tolist())
signals.append(dat['people_mask'].tolist())
signals.append(dat['software_mask'].tolist())
signals = np.array(signals).T
wfdb.wrsamp(dat['labels'], fs=500, units=['mV', 'num', 'num'], sig_name=['ECG', 'visual_mask', 'software_mask'], p_signal=signals, fmt=['16', '16', '16'], write_dir='F:/PhysioBank_Databases/tele/')
wfdb.wrann(dat['labels'], 'atr', sample, symbol,write_dir='F:/PhysioBank_Databases/tele/')