Only add colorbar if the cmap is not the default

2026-07-08 18:24:56 +08:00 · 2015-07-16 21:39:05 -05:00
parent 9fb7676c92
commit 5632c9c1a9
1 changed files with 2 additions and 1 deletions
@@ -148,7 +148,8 @@ def imshow(im, *args, **kwargs):
    kwargs.setdefault('vmin', lo)
    kwargs.setdefault('vmax', hi)
    ax_im = plt.imshow(im, *args, **kwargs)
-    plt.colorbar()
+    if cmap != _default_colormap:
+        plt.colorbar()
    return ax_im

 imread = plt.imread