def test_pac_vec(): """Definition of PAC vectors.""" assert pac_vec() assert pac_vec(fpha=(1, 30, 2, 2), famp=(60, 200, 10, 5)) assert pac_vec(fpha=[1, 2], famp=np.arange(50)) assert pac_vec(fpha=np.array([[2, 4], [5, 7], [9, 10]]), famp=np.array([[30, 60], [60, 90], [100, 200]]).T) assert pac_vec(fpha=[[1, 2], [5, 7]], famp=[60, 150])
def __init__(self, f_pha=[2, 4], f_amp=[60, 200], dcomplex='hilbert', cycle=(3, 6), width=7): # Frequency checking : from tensorpac.utils import pac_vec self._f_pha, self._f_amp = pac_vec(f_pha, f_amp) self._xvec, self._yvec = self.f_pha.mean(1), self.f_amp.mean(1) # Check spectral properties : self._speccheck(dcomplex, cycle, width)
def __init__(self, f_pha=[2, 4], f_amp=[60, 200], dcomplex='hilbert', filt='fir1', cycle=(3, 6), filtorder=3, width=7): # Frequency checking : self._f_pha, self._f_amp = pac_vec(f_pha, f_amp) self._xvec, self._yvec = self.f_pha.mean(1), self.f_amp.mean(1) # Check spectral properties : self._speccheck(filt, dcomplex, filtorder, cycle, width)
avecm = avec.mean(1) plt.subplot(1, 4, nb) plt.vlines(pvecm, -10, 500, color='#ab4642') plt.hlines(avecm, -10, 500, color='slateblue') plt.xlabel('Frequency for phase (hz') plt.ylabel('Frequency for amplitude.mean(1)') plt.title(title) plt.xlim([0, 30]) plt.ylim([60, 200]) plt.figure(figsize=(25, 5)) # 1 - Manual defintion : fpha = [2, 4] famp = [60, 160] pvec1, avec1 = pac_vec(fpha, famp) plot(1, pvec1, avec1, '1 - One frequency band') # 2 - List/tuple/array : fpha = [[2, 4], [5, 7], [8, 13]] famp = ([60, 160], [60, 200]) pvec2, avec2 = pac_vec(fpha, famp) plot(2, pvec2, avec2, 'Manually define several frequency bands') # 3 - (start, end, width, step) : fpha = (1, 30, 2, 1) famp = (60, 200, 10, 5) pvec3, avec3 = pac_vec(fpha, famp) plot(3, pvec3, avec3, 'Use the (start, stop, width, step definition') # 4 - Range :