Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import xarray as xr
- from glob import glob
- paths = glob("/lsdf/kit/imk-tro/projects/MOD/Gruppe_Voigt/TRACMIP_ESGFCOPY/*/%s/%s*/%s/*/*/*/*/*/*"
- % ("ECHAM63", "aquaControl", "mon"))
- for version in set([i.split("/")[-2] for i in paths]):
- ver_paths = [i for i in paths if version in i]
- merged = xr.open_dataset(ver_paths[0], chunks={})
- for path in ver_paths:
- # flatten height coordinate
- ds = xr.open_dataset(path)
- var = [i for i in ds.data_vars if "bnds" not in i][0]
- if "height" in ds.coords:
- ds[var].attrs["height"] = "%s %s" % (ds.height.values, ds.height.units)
- merged[var] = ds[var].chunk({"time" : "auto"})
- # drop height from overall dataset
- if "height" in merged.coords:
- merged = merged.drop("height")
- # convert to zarr
- merged.to_zarr("temp", mode="w", consolidated=True)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement