aboutsummaryrefslogtreecommitdiff
path: root/utils/plot-muons
diff options
context:
space:
mode:
authortlatorre <tlatorre@uchicago.edu>2021-01-03 11:40:47 -0600
committertlatorre <tlatorre@uchicago.edu>2021-01-03 11:40:47 -0600
commit6fc97e22fa2a8cc48a96b2ce22c4d55a8fd2b80f (patch)
treef5bf72de18002e0d126b8e37c628f68ec3926fb6 /utils/plot-muons
parentc71532924c8c2be07026041b7cd62819e90e8c64 (diff)
downloadsddm-6fc97e22fa2a8cc48a96b2ce22c4d55a8fd2b80f.tar.gz
sddm-6fc97e22fa2a8cc48a96b2ce22c4d55a8fd2b80f.tar.bz2
sddm-6fc97e22fa2a8cc48a96b2ce22c4d55a8fd2b80f.zip
cache results from get_events()
Diffstat (limited to 'utils/plot-muons')
-rwxr-xr-xutils/plot-muons5
1 files changed, 1 insertions, 4 deletions
diff --git a/utils/plot-muons b/utils/plot-muons
index 2616ad2..a37f223 100755
--- a/utils/plot-muons
+++ b/utils/plot-muons
@@ -102,11 +102,8 @@ if __name__ == '__main__':
ev = ev[~np.isnan(ev.fmin)]
ev_mc = ev_mc[~np.isnan(ev_mc.fmin)]
- ev = ev.reset_index()
- ev_mc = ev_mc.reset_index()
-
# Set all prompt events in the MC to be muons
- ev_mc.loc[ev_mc.prompt & ev_mc.filename.str.contains("cosmic"),'muon'] = True
+ ev_mc.loc[ev_mc.prompt,'muon'] = True
# First, do basic data cleaning which is done for all events.
ev = ev[ev.signal | ev.muon]