From 83f09a55bd6498be31c3641111f608cb62380d2d Mon Sep 17 00:00:00 2001 From: Neil Rickards Date: Fri, 1 Jan 2021 21:49:09 +0000 Subject: [PATCH] Update 3rd-gc.py Some minor refactoring --- 3rd-gc.py | 25 ++++++++----------------- 1 file changed, 8 insertions(+), 17 deletions(-) diff --git a/3rd-gc.py b/3rd-gc.py index 7b42992..9989610 100644 --- a/3rd-gc.py +++ b/3rd-gc.py @@ -2,40 +2,30 @@ def read_csv(filename): - records = [] with open(filename, 'rt') as fp: reader = csv.reader(fp, delimiter=',') - for ii, row in enumerate(reader): - if ii > 0: - records.append(row) - - return records + next(reader) + return list(reader) codons = read_csv('codon-table-grouped.csv') #print(codons) -c2s= {} -for c in codons: - c2s[c[1]] = c[0] +c2s= {amino: codon for codon, amino in codons} print(c2s) - virvac = read_csv("side-by-side.csv") #print(virvac) matches = 0 -for element in virvac: - - vir = element[1] - vac = element[2] +for (_, vir, vac) in virvac: print(f'{vir} v {vac}, amino: {c2s[vir]} == {c2s[vac]}.') our = vir - if vir[2] == 'G' or vir[2] == 'C': + if vir[2] in ['G','C']: print('codon ended on G or C already, not doing anything') else: @@ -46,7 +36,7 @@ def read_csv(filename): print('amino acid still the same, done!') our = prop else: - print(f'Oops, maino acid changed. Trying C, new candidate {prop}') + print(f'Oops, amino acid changed. Trying C, new candidate {prop}') prop = vir[:2] + "C" if c2s[vir] == c2s[prop]: @@ -55,7 +45,8 @@ def read_csv(filename): if vac == our: print('Matched the vaccine!') - matches +=1 + matches += 1 + else: print('No Match.')