Пример #1
0
def preprocessing():
    patients = []
    with open('states_new.csv', 'rb') as csvfile:
        reader = csv.reader(csvfile, delimiter=',', quotechar='|')
        patient = Patient()
        patient.id = 1
        i = 0
        chain = []
        times = []
        divider = 100
        for row in reader:
            #check whether patient the same one
            if i == 0:
                i += 1
                continue
            if int(row[0]) == patient.id:
                chain.append(int(row[4]))
                times.append(float(row[5]) / divider)
            else:
                #move to next patient
                patient.O = np.array(chain)
                patient.observation_times = np.array(times)
                patient.T_obs = len(times)
                patients.append(patient)
                patient = Patient()
                patient.id = int(row[0])
                chain = [int(row[4])]
                times = [float(row[5]) / divider]
            i += 1

        #setup final patient
        patient.O = np.array(chain)
        patient.observation_times = np.array(times)
        patient.T_obs = len(times)
        patients.append(patient)
        patient = Patient()
        patient.id = int(row[0])
    return patients