Change to just apply to pcolormesh, not pcolor

jklymak · jklymak · commit b6f89237ba6b · 2017-11-10T14:27:40.000-08:00
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -5186,18 +5186,18 @@ def _pcolorargs(funcname, *args, **kw):
             X, Y, C = [np.asanyarray(a) for a in args]
             for a in X, Y:
                 a = cbook.safe_masked_invalid(a)
-                if np.ma.is_masked(a):
+                if funcname == 'pcolormesh' and np.ma.is_masked(a):
                     raise TypeError(
-                        'x and y arguments to %s cannot have '
-                        'non-finite values and cannot be type '
-                        'numpy.ma.core.MaskedArray with masked values'
-                        % funcname)
-            # if it didn't fail the above it is masked but has no
-            # masked values, so strip the mask:
-            if type(X) is np.ma.core.MaskedArray:
-                X = X.data
-            if type(Y) is np.ma.core.MaskedArray:
-                Y = Y.data
+                        'x and y arguments to pcolormesh cannot have '
+                        'non-finite values and or be of type '
+                        'numpy.ma.core.MaskedArray with masked values')
+            if funcname == 'pcolormesh':
+                # if it didn't fail the above it is masked but has no
+                # masked values, so strip the mask for pcolormesh
+                if type(X) is np.ma.core.MaskedArray:
+                    X = X.data
+                if type(Y) is np.ma.core.MaskedArray:
+                    Y = Y.data
             numRows, numCols = C.shape
         else:
             raise TypeError(
