示例#1
0
tot = 0
while su < NUM_SURR:
    if AMPLITUDE:
        sg_amp = SurrogateField()
        sg_amp.copy_field(g_amp)
    sg = SurrogateField()
    sg.copy_field(g)
    if SURR_TYPE == 'MF':
        if AMPLITUDE:
            sg_amp.construct_multifractal_surrogates()
            sg_amp.add_seasonality(mean2, var2, trend2)
        sg.construct_multifractal_surrogates()
        sg.add_seasonality(mean, var, trend)
    elif SURR_TYPE == 'FT':
        if AMPLITUDE:
            sg_amp.construct_fourier_surrogates_spatial()
            sg_amp.add_seasonality(mean2, var2, trend2)
        sg.construct_fourier_surrogates_spatial()
        sg.add_seasonality(mean, var, trend)
    elif SURR_TYPE == 'AR':
        if AMPLITUDE:
            sg_amp.prepare_AR_surrogates()
            sg_amp.construct_surrogates_with_residuals()
            sg_amp.add_seasonality(mean2, var2, trend2)
        sg.prepare_AR_surrogates()
        sg.construct_surrogates_with_residuals()
        sg.add_seasonality(mean[:-1], var[:-1], trend[:-1])
    wave, _, _, _ = wavelet_analysis.continous_wavelet(
        sg.surr_data,
        1,
        True,
daily_var = np.zeros((365,3))
mean, var_data, trend = ts.g.get_seasonality(True)
sg.copy_field(ts.g)

#MF
sg.construct_multifractal_surrogates()
sg.add_seasonality(mean, var_data, trend)

g.data = sg.surr_data.copy()
g.time = sg.time.copy()

_, var_surr_MF, _ = g.get_seasonality(True)

#FT
sg.construct_fourier_surrogates_spatial()
sg.add_seasonality(mean, var_data, trend)

g.data = sg.surr_data.copy()
g.time = sg.time.copy()

_, var_surr_FT, _ = g.get_seasonality(True)

delta = timedelta(days = 1)
d = date(1895,1,1)

for i in range(daily_var.shape[0]):
    ndx = ts.g.find_date_ndx(d)
    daily_var[i,0] = var_data[ndx]
    daily_var[i,1] = var_surr_MF[ndx]
    daily_var[i,2] = var_surr_FT[ndx]