Fix a recursion problem with masked arrays in get_converter introduced in f8344a1

selasley · selasley · commit d6340aba6729 · 2013-08-10T16:38:25.000-04:00
If a masked array with the first element masked is passed in to get_converter, the   converter = self.get_converter(xravel()[0])  call causes a RuntimeError: maximum recursion depth exceeded.  This patch will pass the first unmasked value, or skip the call if all elements are masked.
diff --git a/lib/matplotlib/units.py b/lib/matplotlib/units.py
@@ -134,8 +134,17 @@ def get_converter(self, x):
             converter = self.get(classx)
 
         if isinstance(x, np.ndarray) and x.size:
-            converter = self.get_converter(x.ravel()[0])
-            return converter
+            xravel = x.ravel()
+            try:
+                # pass the first value of x that is not masked back to get_converter
+                if not np.all(xravel.mask):
+                    # some elements are not masked
+                    converter = self.get_converter(xravel[np.argmin(xravel.mask)])
+                    return converter
+            except AttributeError:
+                # not a masked_array
+                converter = self.get_converter(xravel[0])
+                return converter
 
         if converter is None and iterable(x):
             for thisx in x:
