def remove_trials(spktimes, spktrials, spkwaveform, lfp, stimID, remID): spk_mask, trial_mask = RF.make_spk_and_trial_masks(spktrials, stimID, remID) if spk_mask.sum()>0: spktimes = spktimes[~spk_mask] spktrials = spktrials[~spk_mask] spkwaveform = spkwaveform[~spk_mask, :] lfp = lfp[~trial_mask, :] stimID = stimID[~trial_mask, :] return spktimes, spktrials, spkwaveform, lfp, stimID