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]