diff --git a/src/LIB/Python/misc_functions.py b/src/LIB/Python/misc_functions.py index 53577de6ee71fbc12524c19af6daff7657bd5fe9..c5b29258e15a1ce6da9b6681201587958c0c4f16 100644 --- a/src/LIB/Python/misc_functions.py +++ b/src/LIB/Python/misc_functions.py @@ -213,9 +213,11 @@ def comp_altitude2DVar(oneVar3D, orography, ztop, level, n_y, n_x): n_x3D = copy.deepcopy(oneVar3D) n_y3D = copy.deepcopy(oneVar3D) altitude = copy.deepcopy(oneVar3D) - for i in range(len(level)): - n_y3D[i, :] = n_y - n_x3D[i, :] = n_x + for k in range(len(level)): + for j in range(oneVar3D.shape[2]): + n_y3D[k,:,j] = n_y + for i in range(oneVar3D.shape[1]): + n_x3D[k,i,:] = n_x for i in range(oneVar3D.shape[2]): for j in range(oneVar3D.shape[1]): if ztop == 0: