forked from mondejar/ecg-classification
-
Notifications
You must be signed in to change notification settings - Fork 0
/
2csv.py
23 lines (17 loc) · 649 Bytes
/
2csv.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
# this script convert the full dataset mitdb to .csv
from os import listdir, mkdir, system
from os.path import isfile, isdir, join, exists
dir = '/local/scratch/mondejar/dataset/mitdb/'
#Create folder
csv = dir + 'csv'
if not exists(csv):
mkdir(csv)
records = [f for f in listdir(dir) if isfile(join(dir, f)) if(f.find('.dat') != -1)]
#print records
for r in records:
command = 'rdsamp -r ' + r[:-4] + ' -c -H -f 0 -v >' + 'csv/' + r[:-4] + '.csv'
print(command)
system(command)
command_annotations = 'rdann -r ' + r[:-4] +' -f 0 -a atr -v >' + 'csv/' + r[:-4] + 'annotations.txt'
print(command_annotations)
system(command_annotations)