ftag.track_selector#

Classes#

TrackSelector

Apply track selections to a set of tracks stored in a structured numpy array.

Module Contents#

class ftag.track_selector.TrackSelector#

Apply track selections to a set of tracks stored in a structured numpy array.

The array is assumed to have shape (n_jets, n_tracks, n_features). Applying cuts will NaN out the tracks that do not pass the cuts, but leave the shape of the array unchanged.

Parameters:

  • cuts (Cuts) – The cuts to apply to the tracks

  • valid_str (str) – The name of the field in the tracks that indicates whether the track is

cuts: ftag.cuts.Cuts#
valid_str: str = 'valid'#
__call__(tracks: numpy.ndarray) → numpy.ndarray#
_nshared_cut(cut: ftag.cuts.Cut, tracks: numpy.ndarray) → numpy.ndarray#