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):