From b2eeccd470490d7acb4cf2784a571ffd68ebd3cb Mon Sep 17 00:00:00 2001 From: Irena Vankova Date: Thu, 21 Aug 2025 15:17:48 -0600 Subject: [PATCH 01/16] removed map inset --- mpas_analysis/ocean/time_series_antarctic_melt.py | 4 ++-- mpas_analysis/ocean/time_series_ocean_regions.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/mpas_analysis/ocean/time_series_antarctic_melt.py b/mpas_analysis/ocean/time_series_antarctic_melt.py index 65dff877d..14668a109 100644 --- a/mpas_analysis/ocean/time_series_antarctic_melt.py +++ b/mpas_analysis/ocean/time_series_antarctic_melt.py @@ -723,7 +723,7 @@ def run_task(self): # and cartopy doesn't play too well with tight_layout anyway plt.tight_layout() - add_inset(fig, fc, width=2.0, height=2.0) + #add_inset(fig, fc, width=2.0, height=2.0) savefig(outFileName, config) @@ -788,7 +788,7 @@ def run_task(self): # and cartopy doesn't play too well with tight_layout anyway plt.tight_layout() - add_inset(fig, fc, width=2.0, height=2.0) + #add_inset(fig, fc, width=2.0, height=2.0) savefig(outFileName, config) diff --git a/mpas_analysis/ocean/time_series_ocean_regions.py b/mpas_analysis/ocean/time_series_ocean_regions.py index cd758e62d..edfe20c50 100644 --- a/mpas_analysis/ocean/time_series_ocean_regions.py +++ b/mpas_analysis/ocean/time_series_ocean_regions.py @@ -1350,7 +1350,7 @@ def run_task(self): # and cartopy doesn't play too well with tight_layout anyway plt.tight_layout() - add_inset(fig, fc, width=2.0, height=2.0) + #add_inset(fig, fc, width=2.0, height=2.0) savefig(outFileName, config, tight=False) From 4ef8c83c938eb86780e84e958f2b64f2e029640d Mon Sep 17 00:00:00 2001 From: Irena Vankova Date: Thu, 21 Aug 2025 15:56:14 -0600 Subject: [PATCH 02/16] moved inset like in hov --- mpas_analysis/ocean/ocean_regional_profiles.py | 11 ++++++++++- mpas_analysis/ocean/time_series_antarctic_melt.py | 2 ++ 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/mpas_analysis/ocean/ocean_regional_profiles.py b/mpas_analysis/ocean/ocean_regional_profiles.py index 7f7c6b57b..797128e96 100644 --- a/mpas_analysis/ocean/ocean_regional_profiles.py +++ b/mpas_analysis/ocean/ocean_regional_profiles.py @@ -349,6 +349,8 @@ def run_task(self): dsMesh = xr.open_dataset(meshFilename) dsMesh = dsMesh.isel(Time=0) areaCell = dsMesh.areaCell + landIceFloatingMask = dsMesh.landIceFloatingMask.isel(Time=0) + landIceFloatingMask = -1*(landIceFloatingMask-1) nVertLevels = dsMesh.sizes['nVertLevels'] @@ -407,7 +409,14 @@ def run_task(self): prefix = field['prefix'] self.logger.info(' {}'.format(field['titleName'])) - var = dsLocal[variableName].where(vertDepthMask) + var_mpas = dsLocal[variableName] + print('DIMS-------------') + print(var_mpas.dims) + print(landIceFloatingMask.dims) + var_mpas_masked = var_mpas*landIceFloatingMask + print(var_mpas_masked.dims) + var = var_mpas_masked.where(vertDepthMask) + print(var.dims) meanName = '{}_mean'.format(prefix) dsLocal[meanName] = \ diff --git a/mpas_analysis/ocean/time_series_antarctic_melt.py b/mpas_analysis/ocean/time_series_antarctic_melt.py index 14668a109..ee1b09f15 100644 --- a/mpas_analysis/ocean/time_series_antarctic_melt.py +++ b/mpas_analysis/ocean/time_series_antarctic_melt.py @@ -724,6 +724,7 @@ def run_task(self): plt.tight_layout() #add_inset(fig, fc, width=2.0, height=2.0) + add_inset(fig, fc, width=1.0, height=1.0, xbuffer=0.1, ybuffer=0.1) savefig(outFileName, config) @@ -789,6 +790,7 @@ def run_task(self): plt.tight_layout() #add_inset(fig, fc, width=2.0, height=2.0) + add_inset(fig, fc, width=1.0, height=1.0, xbuffer=0.1, ybuffer=0.1) savefig(outFileName, config) From 04d638b2b2c23eb8db2a223e749883cb2f0dffb4 Mon Sep 17 00:00:00 2001 From: Irena Vankova Date: Thu, 21 Aug 2025 16:16:16 -0600 Subject: [PATCH 03/16] landIceFraction instead of landIceFloatingMask --- mpas_analysis/ocean/ocean_regional_profiles.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/mpas_analysis/ocean/ocean_regional_profiles.py b/mpas_analysis/ocean/ocean_regional_profiles.py index 797128e96..3f052f4cf 100644 --- a/mpas_analysis/ocean/ocean_regional_profiles.py +++ b/mpas_analysis/ocean/ocean_regional_profiles.py @@ -349,8 +349,9 @@ def run_task(self): dsMesh = xr.open_dataset(meshFilename) dsMesh = dsMesh.isel(Time=0) areaCell = dsMesh.areaCell - landIceFloatingMask = dsMesh.landIceFloatingMask.isel(Time=0) - landIceFloatingMask = -1*(landIceFloatingMask-1) + landIceFraction = dsMesh.landIceFraction.isel(Time=0) + landIceFraction = xr.where(landIceFraction > 0, 1, landIceFraction) + landIceFraction = -1*(landIceFraction-1) nVertLevels = dsMesh.sizes['nVertLevels'] @@ -412,8 +413,8 @@ def run_task(self): var_mpas = dsLocal[variableName] print('DIMS-------------') print(var_mpas.dims) - print(landIceFloatingMask.dims) - var_mpas_masked = var_mpas*landIceFloatingMask + print(landIceFraction.dims) + var_mpas_masked = var_mpas*landIceFraction print(var_mpas_masked.dims) var = var_mpas_masked.where(vertDepthMask) print(var.dims) From 339013d2e71494e968050cd5b7759341972dbb87 Mon Sep 17 00:00:00 2001 From: Irena Vankova Date: Thu, 21 Aug 2025 16:31:09 -0600 Subject: [PATCH 04/16] landIceFraction no time --- mpas_analysis/ocean/ocean_regional_profiles.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mpas_analysis/ocean/ocean_regional_profiles.py b/mpas_analysis/ocean/ocean_regional_profiles.py index 3f052f4cf..04497add0 100644 --- a/mpas_analysis/ocean/ocean_regional_profiles.py +++ b/mpas_analysis/ocean/ocean_regional_profiles.py @@ -349,7 +349,7 @@ def run_task(self): dsMesh = xr.open_dataset(meshFilename) dsMesh = dsMesh.isel(Time=0) areaCell = dsMesh.areaCell - landIceFraction = dsMesh.landIceFraction.isel(Time=0) + landIceFraction = dsMesh.landIceFraction landIceFraction = xr.where(landIceFraction > 0, 1, landIceFraction) landIceFraction = -1*(landIceFraction-1) From 841e8918cb92056e139b9675f6bdf6b97f8dabd1 Mon Sep 17 00:00:00 2001 From: Irena Vankova Date: Fri, 22 Aug 2025 12:21:50 -0600 Subject: [PATCH 05/16] shift inset --- mpas_analysis/ocean/time_series_antarctic_melt.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mpas_analysis/ocean/time_series_antarctic_melt.py b/mpas_analysis/ocean/time_series_antarctic_melt.py index ee1b09f15..ec56a4a8e 100644 --- a/mpas_analysis/ocean/time_series_antarctic_melt.py +++ b/mpas_analysis/ocean/time_series_antarctic_melt.py @@ -724,7 +724,7 @@ def run_task(self): plt.tight_layout() #add_inset(fig, fc, width=2.0, height=2.0) - add_inset(fig, fc, width=1.0, height=1.0, xbuffer=0.1, ybuffer=0.1) + add_inset(fig, fc, width=1.0, height=1.0, xbuffer=0.3, ybuffer=0.3) savefig(outFileName, config) @@ -790,7 +790,7 @@ def run_task(self): plt.tight_layout() #add_inset(fig, fc, width=2.0, height=2.0) - add_inset(fig, fc, width=1.0, height=1.0, xbuffer=0.1, ybuffer=0.1) + add_inset(fig, fc, width=1.0, height=1.0, xbuffer=0.3, ybuffer=0.3) savefig(outFileName, config) From 52e578425354e4063f6073671982ccfc1770a809 Mon Sep 17 00:00:00 2001 From: Irena Vankova Date: Fri, 22 Aug 2025 12:27:25 -0600 Subject: [PATCH 06/16] GT to Gt --- mpas_analysis/ocean/time_series_antarctic_melt.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mpas_analysis/ocean/time_series_antarctic_melt.py b/mpas_analysis/ocean/time_series_antarctic_melt.py index ec56a4a8e..73a60dee9 100644 --- a/mpas_analysis/ocean/time_series_antarctic_melt.py +++ b/mpas_analysis/ocean/time_series_antarctic_melt.py @@ -342,7 +342,7 @@ def run_task(self): dsOut = xarray.concat(objs=datasets, dim='Time') dsOut['regionNames'] = dsRegionMask.regionNames - dsOut.integratedMeltFlux.attrs['units'] = 'GT a$^{-1}$' + dsOut.integratedMeltFlux.attrs['units'] = 'Gt a$^{-1}$' dsOut.integratedMeltFlux.attrs['description'] = \ 'Integrated melt flux summed over each ice shelf or region' dsOut.meltRates.attrs['units'] = 'm a$^{-1}$' @@ -662,7 +662,7 @@ def run_task(self): suffix = self.iceShelf.replace(' ', '_') xLabel = 'Time (yr)' - yLabel = 'Melt Flux (GT/yr)' + yLabel = 'Melt Flux (Gt/yr)' timeSeries = integratedMeltFlux.isel(nRegions=self.regionIndex) From d75e26851856dd802a38e1b2d34c846d8e518720 Mon Sep 17 00:00:00 2001 From: Irena Vankova Date: Fri, 22 Aug 2025 12:29:20 -0600 Subject: [PATCH 07/16] less inset buffer --- mpas_analysis/ocean/time_series_antarctic_melt.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mpas_analysis/ocean/time_series_antarctic_melt.py b/mpas_analysis/ocean/time_series_antarctic_melt.py index 73a60dee9..4257f158f 100644 --- a/mpas_analysis/ocean/time_series_antarctic_melt.py +++ b/mpas_analysis/ocean/time_series_antarctic_melt.py @@ -724,7 +724,7 @@ def run_task(self): plt.tight_layout() #add_inset(fig, fc, width=2.0, height=2.0) - add_inset(fig, fc, width=1.0, height=1.0, xbuffer=0.3, ybuffer=0.3) + add_inset(fig, fc, width=1.0, height=1.0, xbuffer=0.01, ybuffer=0.01) savefig(outFileName, config) @@ -790,7 +790,7 @@ def run_task(self): plt.tight_layout() #add_inset(fig, fc, width=2.0, height=2.0) - add_inset(fig, fc, width=1.0, height=1.0, xbuffer=0.3, ybuffer=0.3) + add_inset(fig, fc, width=1.0, height=1.0, xbuffer=0.01, ybuffer=0.01) savefig(outFileName, config) From 516841b72fb93efd722279c7381ed13eb3fbddae Mon Sep 17 00:00:00 2001 From: Irena Vankova Date: Fri, 22 Aug 2025 12:50:07 -0600 Subject: [PATCH 08/16] remove print statements --- mpas_analysis/ocean/ocean_regional_profiles.py | 5 ----- mpas_analysis/ocean/time_series_ocean_regions.py | 1 + 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/mpas_analysis/ocean/ocean_regional_profiles.py b/mpas_analysis/ocean/ocean_regional_profiles.py index 04497add0..b5be6444e 100644 --- a/mpas_analysis/ocean/ocean_regional_profiles.py +++ b/mpas_analysis/ocean/ocean_regional_profiles.py @@ -411,13 +411,8 @@ def run_task(self): self.logger.info(' {}'.format(field['titleName'])) var_mpas = dsLocal[variableName] - print('DIMS-------------') - print(var_mpas.dims) - print(landIceFraction.dims) var_mpas_masked = var_mpas*landIceFraction - print(var_mpas_masked.dims) var = var_mpas_masked.where(vertDepthMask) - print(var.dims) meanName = '{}_mean'.format(prefix) dsLocal[meanName] = \ diff --git a/mpas_analysis/ocean/time_series_ocean_regions.py b/mpas_analysis/ocean/time_series_ocean_regions.py index edfe20c50..414af0831 100644 --- a/mpas_analysis/ocean/time_series_ocean_regions.py +++ b/mpas_analysis/ocean/time_series_ocean_regions.py @@ -1351,6 +1351,7 @@ def run_task(self): plt.tight_layout() #add_inset(fig, fc, width=2.0, height=2.0) + add_inset(fig, fc, width=1.0, height=1.0, xbuffer=0.01, ybuffer=0.01) savefig(outFileName, config, tight=False) From effdc971893b296be4d8216624fdae3cb2d883f9 Mon Sep 17 00:00:00 2001 From: Irena Vankova Date: Fri, 22 Aug 2025 13:02:35 -0600 Subject: [PATCH 09/16] inset lower left --- mpas_analysis/ocean/time_series_antarctic_melt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mpas_analysis/ocean/time_series_antarctic_melt.py b/mpas_analysis/ocean/time_series_antarctic_melt.py index 4257f158f..ff460a45c 100644 --- a/mpas_analysis/ocean/time_series_antarctic_melt.py +++ b/mpas_analysis/ocean/time_series_antarctic_melt.py @@ -790,7 +790,7 @@ def run_task(self): plt.tight_layout() #add_inset(fig, fc, width=2.0, height=2.0) - add_inset(fig, fc, width=1.0, height=1.0, xbuffer=0.01, ybuffer=0.01) + add_inset(fig, fc, width=1.0, height=1.0, lowerleft=[0.0, 0.0], xbuffer=0.01, ybuffer=0.01) savefig(outFileName, config) From 0a825de53d0d36437e6bfa98b9ea61b373998486 Mon Sep 17 00:00:00 2001 From: Irena Vankova Date: Fri, 22 Aug 2025 13:08:12 -0600 Subject: [PATCH 10/16] inset lower left 2 --- mpas_analysis/ocean/time_series_antarctic_melt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mpas_analysis/ocean/time_series_antarctic_melt.py b/mpas_analysis/ocean/time_series_antarctic_melt.py index ff460a45c..ad44b05cf 100644 --- a/mpas_analysis/ocean/time_series_antarctic_melt.py +++ b/mpas_analysis/ocean/time_series_antarctic_melt.py @@ -724,7 +724,7 @@ def run_task(self): plt.tight_layout() #add_inset(fig, fc, width=2.0, height=2.0) - add_inset(fig, fc, width=1.0, height=1.0, xbuffer=0.01, ybuffer=0.01) + add_inset(fig, fc, width=1.0, height=1.0, lowerleft=[0.0, 0.0], xbuffer=0.01, ybuffer=0.01) savefig(outFileName, config) From 558993473e6868b78bb0c285134c69eb3e29c4fc Mon Sep 17 00:00:00 2001 From: Irena Vankova Date: Fri, 22 Aug 2025 13:22:33 -0600 Subject: [PATCH 11/16] inset lower left volume integrated --- mpas_analysis/ocean/time_series_ocean_regions.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mpas_analysis/ocean/time_series_ocean_regions.py b/mpas_analysis/ocean/time_series_ocean_regions.py index 414af0831..c164f225f 100644 --- a/mpas_analysis/ocean/time_series_ocean_regions.py +++ b/mpas_analysis/ocean/time_series_ocean_regions.py @@ -1351,7 +1351,7 @@ def run_task(self): plt.tight_layout() #add_inset(fig, fc, width=2.0, height=2.0) - add_inset(fig, fc, width=1.0, height=1.0, xbuffer=0.01, ybuffer=0.01) + add_inset(fig, fc, width=1.0, height=1.0, lowerleft=[0.0, 0.0], xbuffer=0.01, ybuffer=0.01) savefig(outFileName, config, tight=False) From 929940aec1cca46610632a1780d61d3db033f4ad Mon Sep 17 00:00:00 2001 From: irenavankova Date: Mon, 25 Aug 2025 14:26:40 -0600 Subject: [PATCH 12/16] Update mpas_analysis/ocean/time_series_ocean_regions.py Co-authored-by: Xylar Asay-Davis --- mpas_analysis/ocean/time_series_ocean_regions.py | 1 - 1 file changed, 1 deletion(-) diff --git a/mpas_analysis/ocean/time_series_ocean_regions.py b/mpas_analysis/ocean/time_series_ocean_regions.py index c164f225f..c84f0dbc6 100644 --- a/mpas_analysis/ocean/time_series_ocean_regions.py +++ b/mpas_analysis/ocean/time_series_ocean_regions.py @@ -1350,7 +1350,6 @@ def run_task(self): # and cartopy doesn't play too well with tight_layout anyway plt.tight_layout() - #add_inset(fig, fc, width=2.0, height=2.0) add_inset(fig, fc, width=1.0, height=1.0, lowerleft=[0.0, 0.0], xbuffer=0.01, ybuffer=0.01) savefig(outFileName, config, tight=False) From e54dd45ca96d8331924d31c036b52e69cdfd7f7e Mon Sep 17 00:00:00 2001 From: irenavankova Date: Mon, 25 Aug 2025 14:26:51 -0600 Subject: [PATCH 13/16] Update mpas_analysis/ocean/time_series_antarctic_melt.py Co-authored-by: Xylar Asay-Davis --- mpas_analysis/ocean/time_series_antarctic_melt.py | 1 - 1 file changed, 1 deletion(-) diff --git a/mpas_analysis/ocean/time_series_antarctic_melt.py b/mpas_analysis/ocean/time_series_antarctic_melt.py index ad44b05cf..b80f05b0c 100644 --- a/mpas_analysis/ocean/time_series_antarctic_melt.py +++ b/mpas_analysis/ocean/time_series_antarctic_melt.py @@ -789,7 +789,6 @@ def run_task(self): # and cartopy doesn't play too well with tight_layout anyway plt.tight_layout() - #add_inset(fig, fc, width=2.0, height=2.0) add_inset(fig, fc, width=1.0, height=1.0, lowerleft=[0.0, 0.0], xbuffer=0.01, ybuffer=0.01) savefig(outFileName, config) From 1cbacf23d9b80ed76a0e673be845709c3081bd17 Mon Sep 17 00:00:00 2001 From: irenavankova Date: Mon, 25 Aug 2025 14:27:01 -0600 Subject: [PATCH 14/16] Update mpas_analysis/ocean/time_series_antarctic_melt.py Co-authored-by: Xylar Asay-Davis --- mpas_analysis/ocean/time_series_antarctic_melt.py | 1 - 1 file changed, 1 deletion(-) diff --git a/mpas_analysis/ocean/time_series_antarctic_melt.py b/mpas_analysis/ocean/time_series_antarctic_melt.py index b80f05b0c..9764fef20 100644 --- a/mpas_analysis/ocean/time_series_antarctic_melt.py +++ b/mpas_analysis/ocean/time_series_antarctic_melt.py @@ -723,7 +723,6 @@ def run_task(self): # and cartopy doesn't play too well with tight_layout anyway plt.tight_layout() - #add_inset(fig, fc, width=2.0, height=2.0) add_inset(fig, fc, width=1.0, height=1.0, lowerleft=[0.0, 0.0], xbuffer=0.01, ybuffer=0.01) savefig(outFileName, config) From e21b5ebc0570def32a0659602079036e3854da1a Mon Sep 17 00:00:00 2001 From: Irena Vankova Date: Mon, 22 Sep 2025 17:17:03 -0600 Subject: [PATCH 15/16] change to openOceanMask --- mpas_analysis/ocean/ocean_regional_profiles.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/mpas_analysis/ocean/ocean_regional_profiles.py b/mpas_analysis/ocean/ocean_regional_profiles.py index b5be6444e..d8e021041 100644 --- a/mpas_analysis/ocean/ocean_regional_profiles.py +++ b/mpas_analysis/ocean/ocean_regional_profiles.py @@ -349,9 +349,7 @@ def run_task(self): dsMesh = xr.open_dataset(meshFilename) dsMesh = dsMesh.isel(Time=0) areaCell = dsMesh.areaCell - landIceFraction = dsMesh.landIceFraction - landIceFraction = xr.where(landIceFraction > 0, 1, landIceFraction) - landIceFraction = -1*(landIceFraction-1) + openOceanMask = xr.where(dsMesh.landIceMask > 0, 0, 1) nVertLevels = dsMesh.sizes['nVertLevels'] @@ -411,7 +409,7 @@ def run_task(self): self.logger.info(' {}'.format(field['titleName'])) var_mpas = dsLocal[variableName] - var_mpas_masked = var_mpas*landIceFraction + var_mpas_masked = var_mpas * openOceanMask var = var_mpas_masked.where(vertDepthMask) meanName = '{}_mean'.format(prefix) From 45d447f9f4c091175e6665599b395209c7ffc2cb Mon Sep 17 00:00:00 2001 From: Xylar Asay-Davis Date: Wed, 1 Oct 2025 06:25:26 -0500 Subject: [PATCH 16/16] Use open ocean mask in _masked_area_sum() method This ensures that both the area-weighted field (numerator) and the total area (denominator) include the same masking. --- .../ocean/ocean_regional_profiles.py | 30 ++++++++++++------- 1 file changed, 19 insertions(+), 11 deletions(-) diff --git a/mpas_analysis/ocean/ocean_regional_profiles.py b/mpas_analysis/ocean/ocean_regional_profiles.py index d8e021041..5ff8f9485 100644 --- a/mpas_analysis/ocean/ocean_regional_profiles.py +++ b/mpas_analysis/ocean/ocean_regional_profiles.py @@ -383,7 +383,9 @@ def run_task(self): cellMasks = dsRegionMask.regionCellMasks regionNamesVar = dsRegionMask.regionNames - totalArea = self._masked_area_sum(cellMasks, areaCell, vertDepthMask) + totalArea = self._masked_area_sum( + cellMasks, openOceanMask, areaCell, vertDepthMask + ) datasets = [] for timeIndex, fileName in enumerate(inputFiles): @@ -408,18 +410,21 @@ def run_task(self): prefix = field['prefix'] self.logger.info(' {}'.format(field['titleName'])) - var_mpas = dsLocal[variableName] - var_mpas_masked = var_mpas * openOceanMask - var = var_mpas_masked.where(vertDepthMask) + var = dsLocal[variableName].where(vertDepthMask) meanName = '{}_mean'.format(prefix) - dsLocal[meanName] = \ - self._masked_area_sum(cellMasks, areaCell, var) / totalArea + dsLocal[meanName] = ( + self._masked_area_sum( + cellMasks, openOceanMask, areaCell, var + ) / totalArea + ) meanSquaredName = '{}_meanSquared'.format(prefix) - dsLocal[meanSquaredName] = \ - self._masked_area_sum(cellMasks, areaCell, var**2) / \ - totalArea + dsLocal[meanSquaredName] = ( + self._masked_area_sum( + cellMasks, openOceanMask, areaCell, var**2 + ) / totalArea + ) # drop the original variables dsLocal = dsLocal.drop_vars(variableList) @@ -452,12 +457,15 @@ def run_task(self): write_netcdf_with_fill(dsOut, outputFileName) @staticmethod - def _masked_area_sum(cellMasks, areaCell, var): + def _masked_area_sum(cellMasks, openOceanMask, areaCell, var): """sum a variable over the masked areas""" nRegions = cellMasks.sizes['nRegions'] totals = [] for index in range(nRegions): - mask = cellMasks.isel(nRegions=slice(index, index+1)) + mask = ( + cellMasks.isel(nRegions=slice(index, index+1)) * + openOceanMask + ) totals.append((mask * areaCell * var).sum('nCells')) total = xr.concat(totals, 'nRegions')