--- a +++ b/IQVIA/nctid2label.py @@ -0,0 +1,29 @@ +import csv + + +def nctid2label_dict(): + nctid2outcome = dict() + outcome2label = dict() + nctid2label = dict() + with open("trialtrove/outcome2label.txt", 'r') as fin: + lines = fin.readlines() + for line in lines: + outcome = line.split('\t')[0] + label = line.strip().split('\t')[1] + outcome2label[outcome] = label + + with open("trialtrove/trial_outcomes_v1.csv", 'r') as csvfile: + csvreader = list(csv.reader(csvfile))[1:] + for row in csvreader: + nctid = row[0] + outcome = row[1] + nctid2outcome[nctid] = outcome + + for nctid,outcome in nctid2outcome.items(): + nctid2label[nctid] = outcome2label[outcome] + + return nctid2label + +nctid2label = nctid2label_dict() +print(nctid2label) +