Models:
Philip Bordner
philipB/
mimic-sparql
0
Downloads: 1
[ab27bc]: / mimicsql / evaluation / process_mimic_db / utils.py

Download this file

37 lines (31 with data), 898 Bytes 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import os

import csv

import sys



def get_patient_name(data_dir):

    pat_id2name = {}

    file_ = os.path.join(data_dir, 'id2name.csv')

    fp = open(file_, 'r')

    for line in csv.reader(fp, delimiter=','):

        pat_id2name[line[0]] = line[1]

    fp.close()

    return pat_id2name



def read_table(data_dir, data_file):

    out_info = []

    

    file_ = os.path.join(data_dir, data_file)

    fp = open(file_, 'r')

    reader = csv.reader(fp, delimiter=',')

    for line in reader:

        header = line

        break

    for line in reader:

        arr = {}

        for k in range(len(header)):

            arr[header[k]] = line[k]

        out_info.append(arr)

    fp.close()

    return out_info



def show_progress(a, b):

    cc = int(round(100.0*float(a)/float(b)))

    dstr = '[' + '>'*cc + ' '*(100-cc) + ']'

    sys.stdout.write(dstr + str(cc) + '%' +'\r')

    sys.stdout.flush()