remove masked/invalid points after forcing to iterable

phobson · phobson · commit 1b3fb65219e7 · 2016-11-07T21:14:24.000-08:00
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -966,15 +966,15 @@ def hlines(self, y, xmin, xmax, colors='k', linestyles='solid',
         xmin = self.convert_xunits(xmin)
         xmax = self.convert_xunits(xmax)
 
-        y, xmin, xmax = cbook.delete_masked_points(y, xmin, xmax)
-
         if not iterable(y):
             y = [y]
         if not iterable(xmin):
             xmin = [xmin]
         if not iterable(xmax):
             xmax = [xmax]
 
+        y, xmin, xmax = cbook.delete_masked_points(y, xmin, xmax)
+
         y = np.ravel(y)
         xmin = np.resize(xmin, y.shape)
         xmax = np.resize(xmax, y.shape)
@@ -1048,15 +1048,15 @@ def vlines(self, x, ymin, ymax, colors='k', linestyles='solid',
         ymin = self.convert_yunits(ymin)
         ymax = self.convert_yunits(ymax)
 
-        x, ymin, ymax = cbook.delete_masked_points(x, ymin, ymax)
-
         if not iterable(x):
             x = [x]
         if not iterable(ymin):
             ymin = [ymin]
         if not iterable(ymax):
             ymax = [ymax]
 
+        x, ymin, ymax = cbook.delete_masked_points(x, ymin, ymax)
+
         x = np.ravel(x)
         ymin = np.resize(ymin, x.shape)
         ymax = np.resize(ymax, x.shape)
