Skip to content

Commit

Permalink
Add void selection_in_band function
Browse files Browse the repository at this point in the history
  • Loading branch information
bpalmeiro committed Mar 7, 2024
1 parent 5a25ad7 commit 16d65f1
Showing 1 changed file with 33 additions and 0 deletions.
33 changes: 33 additions & 0 deletions invisible_cities/reco/icaro_components.py
Original file line number Diff line number Diff line change
Expand Up @@ -133,4 +133,37 @@ def band_selector_and_check(dst : pd.DataFrame,
return sel_krband


def selection_in_band(z : np.array,
e : np.array,
range_z : Tuple[float, float],
range_e : Tuple[float, float],
nbins_z : int = 50,
nbins_e : int = 100,
nsigma : float = 3.5) ->np.array:
"""
This function returns a mask for the selection of the events that are inside the Kr E vz Z
Parameters
----------
z: np.array
axial (z) values
e: np.array
energy values
range_z: Tuple[np.array, np.array]
Range in Z-axis
range_e: Tuple[np.array, np.array]
Range in Energy-axis
nbins_z: int
Number of bins in Z-axis
nbins_e: int
Number of bins in energy-axis
nsigma: float
Number of sigmas to set the band width
Returns
----------
A mask corresponding to the selection made.
"""

# To be implemented

return [True] * len(z)

0 comments on commit 16d65f1

Please sign in to comment.