Merge pull request #15 from fusion-energy/removing_noise_from_log_plo…

…t_zeros imshow interpolation defaulting to none to remove noise
fusion-energy · Nov 30, 2023 · 7ffea38 · 7ffea38
2 parents 826f277 + ccd8db7
commit 7ffea38
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/openmc_cylindrical_mesh_plotter/core.py b/src/openmc_cylindrical_mesh_plotter/core.py
@@ -159,7 +159,11 @@ def plot_mesh_tally_rz_slice(
         axes.set_xlabel(xlabel)
         axes.set_ylabel(ylabel)
 
-    im = axes.imshow(data, extent=(x_min, x_max, y_min, y_max), **kwargs)
+    # zero values with logscale produce noise / fuzzy on the time but setting interpolation to none solves this
+    default_imshow_kwargs = {"interpolation": "none"}
+    default_imshow_kwargs.update(kwargs)
+
+    im = axes.imshow(data, extent=(x_min, x_max, y_min, y_max), **default_imshow_kwargs)
 
     if colorbar:
         fig.colorbar(im, **colorbar_kwargs)