Simplify expression

mdboom · Feb 9, 2016 · 6df0c67 · 6df0c67
1 parent 4cf6a9e
commit 6df0c67
Showing 1 changed file with 1 addition and 1 deletion.
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):