From 54a38138a7241533d4c3ae4f78c31e99aead0eff Mon Sep 17 00:00:00 2001 From: Andrija Kostic Date: Fri, 9 Jul 2021 18:13:53 +0000 Subject: [PATCH 1/2] A quick fix to make the colorbars nicer and adapt their size to the 2D plots canvas --- src/plot.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/plot.py b/src/plot.py index 6dec7a78..50c01959 100644 --- a/src/plot.py +++ b/src/plot.py @@ -415,11 +415,17 @@ def _plot2D(f, ax, **kwargs): rgb, extent=[0, nx*dx, 0, ny*dy], origin="lower", **norm, **aspect) else: + from mpl_toolkits.axes_grid1 import make_axes_locatable + im = ax.imshow( f.val.T, extent=[0, nx*dx, 0, ny*dy], vmin=kwargs.get("vmin"), vmax=kwargs.get("vmax"), cmap=cmap, origin="lower", **norm, **aspect) - plt.colorbar(im) + + divider = make_axes_locatable(ax) + cax = divider.append_axes("right", size="5%", pad=0.05) + + plt.colorbar(im, cax=cax) _limit_xy(**kwargs) return elif isinstance(dom, (HPSpace, GLSpace)): -- GitLab From 55910e94021dd25edfda95e94c8c8e842015ee34 Mon Sep 17 00:00:00 2001 From: Martin Reinecke Date: Mon, 12 Jul 2021 16:59:20 +0200 Subject: [PATCH 2/2] repair broken indentation --- src/plot.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/plot.py b/src/plot.py index 50c01959..3421abea 100644 --- a/src/plot.py +++ b/src/plot.py @@ -422,10 +422,10 @@ def _plot2D(f, ax, **kwargs): vmin=kwargs.get("vmin"), vmax=kwargs.get("vmax"), cmap=cmap, origin="lower", **norm, **aspect) - divider = make_axes_locatable(ax) - cax = divider.append_axes("right", size="5%", pad=0.05) + divider = make_axes_locatable(ax) + cax = divider.append_axes("right", size="5%", pad=0.05) - plt.colorbar(im, cax=cax) + plt.colorbar(im, cax=cax) _limit_xy(**kwargs) return elif isinstance(dom, (HPSpace, GLSpace)): -- GitLab