diff options
Diffstat (limited to 'utils/plot-muons')
| -rwxr-xr-x | utils/plot-muons | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/utils/plot-muons b/utils/plot-muons index 25e7d24..27d3423 100755 --- a/utils/plot-muons +++ b/utils/plot-muons @@ -75,6 +75,7 @@ if __name__ == '__main__': from sddm.plot_energy import * from sddm.plot import * from sddm import setup_matplotlib + from sddm.utils import correct_energy_bias parser = argparse.ArgumentParser("plot fit results") parser.add_argument("filenames", nargs='+', help="input files") @@ -93,7 +94,9 @@ if __name__ == '__main__': for run, df in rhdr.groupby('run'): evs.append(get_events(df.filename.values, merge_fits=True, nhit_thresh=args.nhit_thresh)) ev = pd.concat(evs) + ev = correct_energy_bias(ev) ev_mc = get_events(args.mc, merge_fits=True) + ev_mc = correct_energy_bias(ev_mc) # Drop events without fits ev = ev[~np.isnan(ev.fmin)] |