Merge pull request matplotlib#26072 from efiring/pcolormesh_masked

Pcolormesh with Gouraud shading: masked arrays

Author: tacaswell

lib/matplotlib/backends/backend_ps.py

@@ -697,6 +697,8 @@ def draw_gouraud_triangle(self, gc, points, colors, trans):
     @_log_if_debug_on
     def draw_gouraud_triangles(self, gc, points, colors, trans):
         assert len(points) == len(colors)
+        if len(points) == 0:
+            return
         assert points.ndim == 3
         assert points.shape[1] == 3
         assert points.shape[2] == 2

lib/matplotlib/collections.py

@@ -2098,6 +2098,10 @@ def _convert_mesh_to_triangles(self, coordinates):
         ], axis=2).reshape((-1, 3, 2))
 
         c = self.get_facecolor().reshape((*coordinates.shape[:2], 4))
+        z = self.get_array()
+        mask = z.mask if np.ma.is_masked(z) else None
+        if mask is not None:
+            c[mask, 3] = np.nan
         c_a = c[:-1, :-1]
         c_b = c[:-1, 1:]
         c_c = c[1:, 1:]
@@ -2109,8 +2113,8 @@ def _convert_mesh_to_triangles(self, coordinates):
             c_c, c_d, c_center,
             c_d, c_a, c_center,
         ], axis=2).reshape((-1, 3, 4))
-
-        return triangles, colors
+        tmask = np.isnan(colors[..., 2, 3])
+        return triangles[~tmask], colors[~tmask]
 
     @artist.allow_rasterization
     def draw(self, renderer):