Merge pull request #27884 from rcomer/rbga-nan

timhoffm · web-flow · commit 3e339a654ec8 · 2024-03-08T10:27:16.000+01:00
FIX: don't copy twice on RGB input
diff --git a/lib/matplotlib/cm.py b/lib/matplotlib/cm.py
@@ -362,7 +362,7 @@ def to_rgba(self, x, alpha=None, bytes=False, norm=True):
                 if xx.dtype.kind == 'f':
                     # If any of R, G, B, or A is nan, set to 0
                     if np.any(nans := np.isnan(x)):
-                        if xx.shape[2] == 4:
+                        if x.shape[2] == 4:
                             xx = xx.copy()
                         xx[np.any(nans, axis=2), :] = 0
 
