import xarray as xr
import sys
import os

# Aggregate and merge the data    
varlist = ['Q2', 'T2', 'PSFC', 'U10', 'V10', 'GLW', 'SWDOWN', 'PREC_ACC_NC', 'RAINNC', 'SNOW_ACC_NC']
idir = '/home/disk/rocinante/DATA/temp/43km/data/unpack/'
odir = '/home/disk/rocinante/DATA/temp/43km/data/yrs/'

yrs = range(2017, 2022)

for v in varlist:
    for yr in yrs:
        print(v, yr)
        ifiles = '{}/{}/wrfout_d4.{}*.nc'.format(idir, v, yr)
        ds = xr.open_mfdataset(ifiles, combine='nested', concat_dim='Time')
        
        out = "{}/{}/".format(odir, v)
        os.makedirs(out, exist_ok=True)
        ds.to_netcdf("{}/{}_{}.nc".format(out, v, yr))