Modify axis aspect and fraction for colorsquare

patniharshit · patniharshit · commit 96af3d30f9d5 · 2017-08-27T15:00:04.000+05:30
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -35,6 +35,7 @@
 from matplotlib.image import FigureImage
 
 import matplotlib.colorbar as cbar
+import matplotlib.colors as mcolors
 
 from matplotlib.axes import Axes, SubplotBase, subplot_class_factory
 from matplotlib.blocking_input import BlockingMouseInput, BlockingKeyMouseInput
@@ -1834,6 +1835,9 @@ def colorbar(self, mappable, cax=None, ax=None, use_gridspec=True, **kw):
         # Store the value of gca so that we can set it back later on.
         current_ax = self.gca()
 
+        if isinstance(mappable.norm, mcolors.BivariateNorm):
+            kw['fraction'] = 0.30
+            kw['aspect'] = 1
         if cax is None:
             if use_gridspec and isinstance(ax, SubplotBase):
                 cax, kw = cbar.make_axes_gridspec(ax, **kw)
