merge_yearly_data.py
36f77c69a22aaw_solar

merge_yearly_data.py
View Options

			import numpy as np
			import xarray as xr
			import util
			import os

			path = "/work/hyenergy/output/meteo/datasets"
			ds = lambda month : ("2001-%02d_all_SIS_4D/RF500/query/query_locs_13d_250" %month)

			prediction = lambda m : os.path.join(path, ds(m), "prediction.nc")
			model_unc = lambda m : os.path.join(path, ds(m), "model_unc.nc" )
			data_unc = lambda m : os.path.join(path, ds(m), "data_unc.nc" )

			target_name = "2001_all_SIS_4D_"

			pred_tgt = os.path.join(path, target_name + "prediction.nc")
			munc_tgt = os.path.join(path, target_name + "model_unc.nc" )
			dunc_tgt = os.path.join(path, target_name + "data_unc.nc" )

			print("\nMerging prediction ...")
			tt = util.Timer()
			util.merge_models(prediction, pred_tgt)
			print("Finished merging prediction")
			tt.restart()

			print("\nMerging model uncertainty ...")
			util.merge_models(model_unc , munc_tgt)
			print("Finished merging model uncertainty")
			tt.restart()

			print("\nMerging data uncertainty ...")
			util.merge_models(data_unc , dunc_tgt)
			print("Finished merging data uncertainty")
			tt.stop()