Updates based on feedback from @anntzer

trygvrad · trygvrad · commit 73713e7ea2c9 · 2025-05-04T11:12:30.000+02:00
diff --git a/lib/matplotlib/colors.py b/lib/matplotlib/colors.py
@@ -3279,7 +3279,8 @@ def __init__(self, norms, vmin=None, vmax=None, clip=False):
         self.vmax = vmax
         self.clip = clip
 
-        [n.callbacks.connect('changed', self._changed) for n in self._norms]
+        for n in self._norms:
+            n.callbacks.connect('changed', self._changed)
 
     @property
     def n_input(self):
@@ -3299,11 +3300,7 @@ def vmin(self):
 
     @vmin.setter
     def vmin(self, value):
-        if not np.iterable(value):
-            value = [value]*self.n_input
-        if len(value) != self.n_input:
-            raise ValueError(f"Invalid vmin for `MultiNorm` with {self.n_input}"
-                             " inputs.")
+        value = np.broadcast_to(value, self.n_input)
         with self.callbacks.blocked():
             for i, v in enumerate(value):
                 if v is not None:
@@ -3316,11 +3313,7 @@ def vmax(self):
 
     @vmax.setter
     def vmax(self, value):
-        if not np.iterable(value):
-            value = [value]*self.n_input
-        if len(value) != self.n_input:
-            raise ValueError(f"Invalid vmax for `MultiNorm` with {self.n_input}"
-                             " inputs.")
+        value = np.broadcast_to(value, self.n_input)
         with self.callbacks.blocked():
             for i, v in enumerate(value):
                 if v is not None:
@@ -3333,8 +3326,7 @@ def clip(self):
 
     @clip.setter
     def clip(self, value):
-        if not np.iterable(value):
-            value = [value]*self.n_input
+        value = np.broadcast_to(value, self.n_input)
         with self.callbacks.blocked():
             for i, v in enumerate(value):
                 if v is not None:
