diff options
| -rwxr-xr-x | utils/plot-muons | 16 |
1 files changed, 11 insertions, 5 deletions
diff --git a/utils/plot-muons b/utils/plot-muons index 2a6e0a2..a55c1ba 100755 --- a/utils/plot-muons +++ b/utils/plot-muons @@ -182,6 +182,12 @@ if __name__ == '__main__': stopping_muons = pd.merge(ev[ev.muon & ev.stopping_muon],michel,left_on=['run','gtid'],right_on=['run','muon_gtid'],suffixes=('','_michel')) stopping_muons_mc = pd.merge(ev_mc[ev_mc.muon & ev_mc.stopping_muon],michel_mc,left_on=['run','gtid'],right_on=['run','muon_gtid'],suffixes=('','_michel')) + stopping_muons = stopping_muons[stopping_muons.nhit_cal < 4000] + stopping_muons_mc = stopping_muons_mc[stopping_muons_mc.nhit_cal < 4000] + + stopping_muons = stopping_muons[stopping_muons.udotr < -0.5] + stopping_muons_mc = stopping_muons_mc[stopping_muons_mc.nhit_cal < -0.5] + if len(stopping_muons): # project muon to PSUP stopping_muons['dx'] = stopping_muons.apply(get_dx,axis=1) @@ -189,14 +195,14 @@ if __name__ == '__main__': # energy based on distance travelled stopping_muons['T_dx'] = dx_to_energy(stopping_muons.dx) stopping_muons_mc['T_dx'] = dx_to_energy(stopping_muons_mc.dx) - stopping_muons['dT'] = stopping_muons['energy1'] - stopping_muons['T_dx'] - stopping_muons_mc['dT'] = stopping_muons_mc['energy1'] - stopping_muons_mc['T_dx'] + stopping_muons['dT'] = stopping_muons['ke'] - stopping_muons['T_dx'] + stopping_muons_mc['dT'] = stopping_muons_mc['ke'] - stopping_muons_mc['T_dx'] - print(stopping_muons[['run','gtid','energy1','T_dx','dT','gtid_michel','r_michel','ftp_r_michel','id','r']]) + print(stopping_muons[['run','gtid','ke','T_dx','dT','gtid_michel','r_michel','ftp_r_michel','id','r']]) fig = plt.figure() - plt.hist((stopping_muons['energy1']-stopping_muons['T_dx'])*100/stopping_muons['T_dx'], bins=np.linspace(-100,100,200), histtype='step', color='C0', label="Data") - plt.hist((stopping_muons_mc['energy1']-stopping_muons_mc['T_dx'])*100/stopping_muons_mc['T_dx'], bins=np.linspace(-100,100,200), histtype='step', color='C1', label="Monte Carlo") + plt.hist((stopping_muons['ke']-stopping_muons['T_dx'])*100/stopping_muons['T_dx'], bins=np.linspace(-100,100,200), histtype='step', color='C0', label="Data") + plt.hist((stopping_muons_mc['ke']-stopping_muons_mc['T_dx'])*100/stopping_muons_mc['T_dx'], bins=np.linspace(-100,100,200), histtype='step', color='C1', label="Monte Carlo") plt.legend() despine(fig,trim=True) plt.xlabel("Fractional energy difference (\%)") |