space varying vertical coordinates of model data— writing netCDF file (xarray)

Question

Consider using a numerical flow model to simulate a simple 1D advection-diffusion case, in hydraulic engineering, e.g.: evolution of salt concentration (Cs). The domain has