From 17e6899bcc57a5d24021cd4b3a9427df2e5628b9 Mon Sep 17 00:00:00 2001 From: tlatorre Date: Fri, 9 Oct 2020 12:55:51 -0500 Subject: only include Monte Carlo files for the runs we are analyzing --- utils/chi2 | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/utils/chi2 b/utils/chi2 index be99626..6764dd5 100755 --- a/utils/chi2 +++ b/utils/chi2 @@ -482,9 +482,18 @@ if __name__ == '__main__': #muon = ev[ev.muon & ev.prompt & ~ev.atm] #muon_atm = ev[ev.muon & ev.prompt & ev.atm] + if (~rhdr.run.isin(ev_mc.run)).any(): + print_warning("Error! The following runs have no Monte Carlo: %s" % \ + rhdr.run[~rhdr.run.isin(ev_mc.run)].values) + sys.exit(1) + + ev_mc = ev_mc[ev_mc.run.isin(rhdr.run)] + data_mc = ev_mc[ev_mc.signal & ev_mc.prompt & ~ev_mc.atm] data_atm_mc = ev_mc[ev_mc.signal & ev_mc.prompt & ev_mc.atm] + data_mc = ev_mc[ev_mc.signal & ev_mc.prompt & ~ev_mc.atm] + bins = np.logspace(np.log10(20),np.log10(10e3),21) atmo_scale_factor = 100.0 -- cgit