Merge pull request matplotlib#23354 from oscargus/listcomprehensions

MNT: Use list comprehension

Author: QuLogic

lib/matplotlib/collections.py

@@ -1232,12 +1232,8 @@ def set_verts_and_codes(self, verts, codes):
         if len(verts) != len(codes):
             raise ValueError("'codes' must be a 1D list or array "
                              "with the same length of 'verts'")
-        self._paths = []
-        for xy, cds in zip(verts, codes):
-            if len(xy):
-                self._paths.append(mpath.Path(xy, cds))
-            else:
-                self._paths.append(mpath.Path(xy))
+        self._paths = [mpath.Path(xy, cds) if len(xy) else mpath.Path(xy)
+                       for xy, cds in zip(verts, codes)]
         self.stale = True
 
 
@@ -1427,14 +1423,10 @@ def __init__(self, segments,  # Can be None.
     def set_segments(self, segments):
         if segments is None:
             return
-        _segments = []
-
-        for seg in segments:
-            if not isinstance(seg, np.ma.MaskedArray):
-                seg = np.asarray(seg, float)
-            _segments.append(seg)
 
-        self._paths = [mpath.Path(_seg) for _seg in _segments]
+        self._paths = [mpath.Path(seg) if isinstance(seg, np.ma.MaskedArray)
+                       else mpath.Path(np.asarray(seg, float))
+                       for seg in segments]
         self.stale = True
 
     set_verts = set_segments  # for compatibility with PolyCollection

lib/matplotlib/contour.py

@@ -531,9 +531,7 @@ def add_label_near(self, x, y, inline=True, inline_spacing=5,
             paths.pop(segmin)
 
             # Add paths if not empty or single point
-            for n in nlc:
-                if len(n) > 1:
-                    paths.append(mpath.Path(n))
+            paths.extend([mpath.Path(n) for n in nlc if len(n) > 1])
 
     def pop_label(self, index=-1):
         """Defaults to removing last label, but any index can be supplied"""

lib/matplotlib/patches.py

@@ -3165,11 +3165,9 @@ def __call__(self, path, mutation_size, linewidth,
                                                         mutation_size,
                                                         linewidth)
                 if np.iterable(fillable):
-                    path_list = []
-                    for p in path_mutated:
-                        # Restore the height
-                        path_list.append(
-                            Path(p.vertices * [1, aspect_ratio], p.codes))
+                    # Restore the height
+                    path_list = [Path(p.vertices * [1, aspect_ratio], p.codes)
+                                 for p in path_mutated]
                     return path_list, fillable
                 else:
                     return path_mutated, fillable