diff --git a/diagonals/regmean.py b/diagonals/regmean.py index 0523d0b7b28767b3a7570157dace51c449fbd24b..03a7802f2e8339effcfa69b3815eb82ddc3c7c03 100644 --- a/diagonals/regmean.py +++ b/diagonals/regmean.py @@ -175,19 +175,14 @@ def _compute_regmean_levels_cpu(var, basins, volume): regmean_total: float32 List containing regional mean at every depth level for variable var. """ - times = var.shape[0] - levs = var.shape[1] regmean_total = {} for basin, mask in basins.items(): - regmean = np.empty((times, levs)) weights = _compute_weights_3d(mask, volume) - for time in range(times): - for lev in range(levs): - regmean = da.ma.average( - var, - axis=(2, 3), - weights=da.broadcast_to(weights, var.shape) - ) + regmean = da.ma.average( + var, + axis=(2, 3), + weights=da.broadcast_to(weights, var.shape) + ) regmean_total[basin] = regmean return regmean_total