get_total_unc.py
b9297dda7962aw_solar

get_total_unc.py
View Options

			import numpy as np
			import xarray as xr
			import pandas as pd

			import os
			import time
			import util
			import sys


			VAR = sys.argv[1]

			## Load data
			gt_path = '/work/hyenergy/output/solar_potential/physical_potential/results'

			dat_unc = xr.open_mfdataset(os.path.join(gt_path, VAR + '_dhm25_200_CH_ELM50_1000_CPU_data_unc.nc'), chunks = {'x' : 100, 'y' : 100})
			mod_unc = xr.open_mfdataset(os.path.join(gt_path, VAR + '_dhm25_200_CH_ELM50_1000_CPU_model_unc.nc'), chunks = {'x' : 100, 'y' : 100})

			tot_unc = xr.ufuncs.sqrt(dat_unc2 + mod_unc2)

			fp = os.path.join(gt_path, VAR + '_dhm25_200_CH_ELM50_1000_CPU_total_unc.nc')
			print('computed total uncertainty; now saving to file %s' %fp)
			tot_unc.to_netcdf(fp)