copy mask locally

Author: Yi Wei

lib/matplotlib/axes/_axes.py

@@ -5740,9 +5740,6 @@ def imshow(self, X, cmap=None, norm=None, *, aspect=None,
         self.add_image(im)
         return im
 
-    def _is_unwritable(self, c):
-        return np.ma.is_masked(c) and not c.mask.flags.writeable
-
     def _pcolorargs(self, funcname, *args, shading='auto', **kwargs):
         # - create X and Y if not present;
         # - reshape X and Y as needed if they are 1-D;
@@ -5767,8 +5764,7 @@ def _pcolorargs(self, funcname, *args, shading='auto', **kwargs):
             else:
                 X, Y = np.meshgrid(np.arange(ncols + 1), np.arange(nrows + 1))
                 shading = 'flat'
-            copy = self._is_unwritable(C)
-            C = cbook.safe_masked_invalid(C, copy=copy)
+            C = cbook.safe_masked_invalid(C, copy=True)
             return X, Y, C, shading
 
         if len(args) == 3:
@@ -5857,8 +5853,7 @@ def _interp_grid(X):
                     Y = _interp_grid(Y.T).T
                 shading = 'flat'
 
-        copy = self._is_unwritable(C)
-        C = cbook.safe_masked_invalid(C, copy=copy)
+        C = cbook.safe_masked_invalid(C, copy=True)
         return X, Y, C, shading
 
     @_preprocess_data()