FIX: Support RGB(A) color arrays for pcolor

greglucas · greglucas · commit ccb3966c2a04 · 2023-04-26T08:53:18.000-06:00
This updates the masked handling to support the 3-dimensional
RGB(A) case if someone inputs those directly rather than mapped
colors.
diff --git a/lib/matplotlib/collections.py b/lib/matplotlib/collections.py
@@ -2252,7 +2252,11 @@ def __init__(self, coordinates, **kwargs):
                 mask[0:-1, 1:] | mask[1:, 0:-1])
         # Take account of the array data too
         C = kwargs.get("array", np.empty(coordinates.shape[:2]))
-        mask |= np.ma.getmaskarray(C)
+        if C.ndim == 3:
+            # RGB(A) case
+            mask |= np.any(np.ma.getmaskarray(C), axis=-1)
+        else:
+            mask |= np.ma.getmaskarray(C)
 
         unmask = ~mask
         self._valid_polys = unmask.ravel()
