FIX: make changing norm reset locator and formatter

jklymak · jklymak · commit 030016125b6d · 2019-01-19T21:17:34.000-08:00
diff --git a/lib/matplotlib/colorbar.py b/lib/matplotlib/colorbar.py
@@ -531,6 +531,8 @@ def update_ticks(self):
         # get the locator and formatter.  Defaults to
         # self.locator if not None..
         locator, formatter = self._get_ticker_locator_formatter()
+        self.locator = locator
+        self.formatter = formatter
         if self.orientation == 'vertical':
             long_axis, short_axis = ax.yaxis, ax.xaxis
         else:
@@ -1140,11 +1142,16 @@ def update_normal(self, mappable):
         Update solid patches, lines, etc.
 
         Unlike `.update_bruteforce`, this does not clear the axes.  This is
-        meant to be called when the image or contour plot to which this
-        colorbar belongs changes.
+        meant to be called when the norm of the image or contour plot to which
+        this colorbar belongs changes.
+
+        This resets the locator and formatter for the axis, so if these
+        have been customized, they will need to be customized again.
         """
 
         self.norm = self.mappable.norm
+        self.locator = None
+        self.formatter = None
         self.draw_all()
         if isinstance(self.mappable, contour.ContourSet):
             CS = self.mappable
@@ -1166,6 +1173,10 @@ def update_bruteforce(self, mappable):
         # properties have been changed by methods other than the
         # colorbar methods, those changes will be lost.
         self.ax.cla()
+        self.locator = None
+        self.formatter = None
+        self.mappable = mappable
+        
         # clearing the axes will delete outline, patch, solids, and lines:
         self.outline = None
         self.patch = None
