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