#!/usr/bin/env python
# Wenchang Yang (wenchang@princeton.edu)
# Mon Aug 12 16:08:03 EDT 2019

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

ens = range(1, 13+1)
TS = []
for en in ens:
    print(en, end='; ')
    ifile = f'/tigress/wenchang/data/cesm/LME/TS/b.e11.BLMTRC5CN.f19_g16.{en:03d}.cam.h0.TS.????01-????12.nc'
    ds = xr.open_mfdataset(ifile)
    TS.append( ds.TS.assign_coords(time=ds.indexes['time'].shift(-1, 'MS')).load() )
print()
TS = xr.concat(TS, dim=pd.Index(ens, name='en'))
TS.to_dataset(name='TS').to_netcdf('LME_TS.nc')