Merge pull request matplotlib#2290 from selasley/master

mdboom · mdboom · commit 51595a1293fb · 2013-08-12T08:25:47.000-07:00
Fix a recursion problem with masked arrays in get_converter
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -584,6 +584,8 @@ def test_fill_between_interpolate():
 
     # Test support for masked arrays.
     y2 = np.ma.masked_greater(y2, 1.0)
+    # Test that plotting works for masked arrays with the first element masked
+    y2[0] = np.ma.masked
     ax1 = fig.add_subplot(212, sharex=ax)
     ax1.plot(x, y1, x, y2, color='black')
     ax1.fill_between(x, y1, y2, where=y2>=y1, facecolor='green', interpolate=True)
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:
