error = {} rmsedata = {} amedata = {} for mem in mem_list: rmseloc = {} ameloc = {} for loc in loc_list: rmse = [] ame = [] for wgt in wgt_list: filename = 'h3dw%2.2dl%2.2dm%2.2d/output_analy.txt' % ( wgt, loc, mem) # analysis analy = ev.get_data(filename) # get errors rms, am = ev.errors(truth, analy) print('%s : RMS = %3.4f , AM = %3.4f' % (filename, rms, am)) rmse.append(rms) ame.append(am) rmseloc[str(loc)] = rmse ameloc[str(loc)] = ame rmsedata[str(mem)] = rmseloc amedata[str(mem)] = ameloc error['rmse'] = rmsedata error['ame'] = amedata with open(jsonfile, 'w') as f: json.dump(error, f) if plot_fig: import matplotlib.pyplot as plt