Fix 2d smoothing to work with masked profiles

actris-cloudnet · Feb 25, 2024 · 1a337a1 · 1a337a1
1 parent 8d959c8
commit 1a337a1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cloudnetpy/plotting/plotting.py b/cloudnetpy/plotting/plotting.py
@@ -486,7 +486,9 @@ def _plot_mesh_data(self, figure_data: FigureData) -> None:
             sigma_units = calc_sigma_units(
                 figure_data.time, alt * 1e3, sigma_minutes=duration, sigma_metres=0
             )
-            self._data = uniform_filter(self._data, sigma_units)
+            valid_time_ind = ~np.all(self._data.mask, axis=1)
+            smoothed_data = uniform_filter(self._data[valid_time_ind, :], sigma_units)
+            self._data[valid_time_ind, :] = smoothed_data
 
         image = self._ax.pcolorfast(
             figure_data.time_including_gaps,