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.

cuts

The cuts to apply to the tracks

Type:

Cuts

valid_str

The name of the field in the tracks that indicates whether the track is

Type:

str, optional

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