From aaae0ac03bd2df2f4e551d4de0d9eb5a1c9a5bd0 Mon Sep 17 00:00:00 2001 From: simbilod <46427609+simbilod@users.noreply.github.com> Date: Sun, 26 Dec 2021 17:41:54 -0500 Subject: [PATCH] level parameter for contour plot calls epsilon data (#1869) --- python/visualization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/visualization.py b/python/visualization.py index 1e17f6a08..647882398 100644 --- a/python/visualization.py +++ b/python/visualization.py @@ -402,7 +402,7 @@ def plot_eps(sim, ax, output_plane=None, eps_parameters=None, frequency=None): if mp.am_master(): if eps_parameters['contour']: - ax.contour(eps_data, 0, colors='black', origin='upper', extent=extent, linewidths=eps_parameters['contour_linewidth']) + ax.contour(eps_data, 0, levels=np.unique(eps_data), colors='black', origin='upper', extent=extent, linewidths=eps_parameters['contour_linewidth']) else: ax.imshow(eps_data, extent=extent, **filter_dict(eps_parameters, ax.imshow)) ax.set_xlabel(xlabel)