From e6635aa8b3f1d475642034f9abd58531b524f73d Mon Sep 17 00:00:00 2001 From: Noe Brucy Date: Thu, 11 Feb 2021 11:35:05 +0100 Subject: [PATCH] [postprocessor] use nanmean for radial averages --- postprocessor.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/postprocessor.py b/postprocessor.py index 0783270..b50426d 100644 --- a/postprocessor.py +++ b/postprocessor.py @@ -1005,12 +1005,13 @@ class PostProcessor(HDF5Container): # mean of all the cells in the bin mean_bin = np.zeros(len(radial_bins) - 1) for j in range(len(radial_bins) - 1): + mask_bin = bins_on_map == j if mass_weighted: - weight = coldens[bins_on_map == j] - mean_bin[j] = np.mean(dmap[bins_on_map == j] * weight) + weight = coldens[mask_bin] + mean_bin[j] = np.nanmean(dmap[mask_bin] * weight) mean_bin[j] = mean_bin[j] / np.mean(weight) else: - mean_bin[j] = np.mean(dmap[bins_on_map == j]) + mean_bin[j] = np.nanmean(dmap[mask_bin]) return mean_bin def _rad_avg_map(self, name, ax_los="z", mass_weighted=False):