replay_trajectory_classification.core.mask

mask(value: ndarray, is_track_interior: ndarray) → ndarray

Set bins that are not part of the track to NaN.

Parameters:

• value (np.ndarray, shape (..., n_bins))

• is_track_interior (np.ndarray, shape (n_bins,))

Returns:

masked_value

Return type:

np.ndarray