diff options
| -rw-r--r-- | utils/sddm/plot_energy.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/utils/sddm/plot_energy.py b/utils/sddm/plot_energy.py index 106c2f9..67849b5 100644 --- a/utils/sddm/plot_energy.py +++ b/utils/sddm/plot_energy.py @@ -453,9 +453,11 @@ def burst_cut(ev): The idea here is to cut instrumentals which somehow make it past the data cleaning cuts. """ + ev['prompt_50'] = (ev.nhit >= 100) + ev.loc[ev.prompt_50,'prompt_50'] &= np.concatenate(([True],np.diff(ev[ev.prompt_50].gtr.values) > 50e6)) burst = (ev.nhit >= 100) burst.loc[burst] &= np.concatenate(([True],np.diff(ev[burst].gtr.values) > BURST_WINDOW)) - return ev.groupby(burst.cumsum()).filter(lambda ev: len(ev[ev.prompt]) <= BURST_MAX_EVENTS).reset_index() + return ev.groupby(burst.cumsum()).filter(lambda ev: len(ev[ev.prompt_50]) <= BURST_MAX_EVENTS).reset_index() def get_events(filenames, merge_fits=False, nhit_thresh=None, mc=False): ev = pd.concat([read_hdf(filename, "ev").assign(filename=filename) for filename in filenames],ignore_index=True) |