From 351f892a84d6c64d1d5cbbf3abea5bc89485576a Mon Sep 17 00:00:00 2001 From: tlatorre Date: Tue, 17 Nov 2020 08:59:37 -0600 Subject: update burst cut --- utils/sddm/plot_energy.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'utils') 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) -- cgit