add masked array support to fill_between

jdh2358 · jdh2358 · commit 71dc0e6effe6 · 2009-04-29T16:07:27.000Z
svn path=/trunk/matplotlib/; revision=7070
diff --git a/examples/user_interfaces/embedding_in_gtk.py b/examples/user_interfaces/embedding_in_gtk.py
@@ -10,8 +10,8 @@
 from numpy import arange, sin, pi
 
 # uncomment to select /GTK/GTKAgg/GTKCairo
-from matplotlib.backends.backend_gtk import FigureCanvasGTK as FigureCanvas
-#from matplotlib.backends.backend_gtkagg import FigureCanvasGTKAgg as FigureCanvas
+#from matplotlib.backends.backend_gtk import FigureCanvasGTK as FigureCanvas
+from matplotlib.backends.backend_gtkagg import FigureCanvasGTKAgg as FigureCanvas
 #from matplotlib.backends.backend_gtkcairo import FigureCanvasGTKCairo as FigureCanvas
 
 
diff --git a/examples/user_interfaces/embedding_in_gtk2.py b/examples/user_interfaces/embedding_in_gtk2.py
@@ -9,8 +9,8 @@
 from numpy import arange, sin, pi
 
 # uncomment to select /GTK/GTKAgg/GTKCairo
-from matplotlib.backends.backend_gtk import FigureCanvasGTK as FigureCanvas
-#from matplotlib.backends.backend_gtkagg import FigureCanvasGTKAgg as FigureCanvas
+#from matplotlib.backends.backend_gtk import FigureCanvasGTK as FigureCanvas
+from matplotlib.backends.backend_gtkagg import FigureCanvasGTKAgg as FigureCanvas
 #from matplotlib.backends.backend_gtkcairo import FigureCanvasGTKCairo as FigureCanvas
 
 # or NavigationToolbar for classic
diff --git a/lib/matplotlib/axes.py b/lib/matplotlib/axes.py
@@ -5832,6 +5832,26 @@ def fill_between(self, x, y1, y2=0, where=None, **kwargs):
         self._process_unit_info(xdata=x, ydata=y1, kwargs=kwargs)
         self._process_unit_info(ydata=y2)
 
+        if where is None:
+            where = np.ones(len(x), np.bool)
+        else:
+            where = np.asarray(where)
+
+        maskedx = isinstance(x, np.ma.MaskedArray)
+        maskedy1 = isinstance(y1, np.ma.MaskedArray)
+        maskedy2 = isinstance(y2, np.ma.MaskedArray)
+
+        if (maskedx or maskedy1 or maskedy2):
+            if maskedx:
+                where = where & (~x.mask)
+
+            if maskedy1:
+                where = where & (~y1.mask)
+
+            if maskedy2:
+                where = where & (~y2.mask)
+
+
         # Convert the arrays so we can work with them
         x = np.asarray(self.convert_xunits(x))
         y1 = np.asarray(self.convert_yunits(y1))
@@ -5843,10 +5863,7 @@ def fill_between(self, x, y1, y2=0, where=None, **kwargs):
         if not cbook.iterable(y2):
             y2 = np.ones_like(x)*y2
 
-        if where is None:
-            where = np.ones(len(x), np.bool)
 
-        where = np.asarray(where)
         assert( (len(x)==len(y1)) and (len(x)==len(y2)) and len(x)==len(where))
 
         polys = []
