import numpy as np
import netCDF4

x = netCDF4.Dataset("/datasets/work/D61_CPSR_A_INPUT/ERA5_global/press_levels_201501.nc")["z"][:]
x = np.moveaxes(x,1,-1)

topo = netCDF4.Dataset("/datasets/work/D61_CPSR_A_INPUT/ERA5_global/ERA5_orography.nc")['z'][:]
topo = np.tile(topo, (x.shape[0],1,1,1))


x = np.concatenate((x,topo), axis=3)