Simplify expression

mdboom · mdboom · commit 34d6cc8efc73 · 2016-02-17T09:47:56.000-05:00
diff --git a/lib/matplotlib/cm.py b/lib/matplotlib/cm.py
@@ -267,7 +267,7 @@ def to_rgba(self, x, alpha=None, bytes=False, norm=True):
         # For floating-point greyscale images, we treat negative as
         # transparent so we copy that over to the alpha channel
         if x.ndim == 2 and x.dtype.kind == 'f':
-            rgba[:, :, 3] = np.where(x < 0.0, 0, rgba[:, :, 3])
+            rgba[:, :, 3][x < 0.0] = 0
         return rgba
 
     def set_array(self, A):
