Create new function to format zdata

blink1073 · blink1073 · commit 88b56be198c4 · 2015-03-17T15:56:21.000-05:00
diff --git a/lib/matplotlib/artist.py b/lib/matplotlib/artist.py
@@ -909,6 +909,21 @@ def get_zdata(self, event):
         """
         return ''
 
+    def format_zdata(self, z):
+        """
+        Return *z* string formatted.
+        """
+        try:
+            is_int = isinstance(z[0], int)
+        except (TypeError, IndexError):
+            is_int = isinstance(z, int)
+            z = [z]
+        if is_int:
+            z = ', '.join('{:0}'.format(item) for item in z)
+        else:
+            z = ', '.join('{:0.3g}'.format(item) for item in z)
+        return z
+
 
 class ArtistInspector(object):
     """
diff --git a/lib/matplotlib/image.py b/lib/matplotlib/image.py
@@ -696,11 +696,7 @@ def get_zdata(self, event):
         y, x = event.ydata, event.xdata
         i, j = trans.transform_point([y, x]).astype(int)
         z = arr[i, j]
-        if z.size > 1:
-            # Override default numpy formatting for this specific case.
-            # Bad idea?
-            z = ', '.join('{:0.3g}'.format(item) for item in z)
-        return 'z=%s' % z
+        return 'z=%s' % self.format_zdata(z)
 
 
 class NonUniformImage(AxesImage):
