Diff of /2csv.py [000000] .. [4d064f]

Switch to side-by-side view

--- a
+++ b/2csv.py
@@ -0,0 +1,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)