Merge pull request #16599 from anntzer/dolphin

timhoffm · web-flow · commit 38ae52b2eb7e · 2020-02-28T23:15:26.000+01:00
Cleanup dolphin example.
diff --git a/examples/shapes_and_collections/dolphin.py b/examples/shapes_and_collections/dolphin.py
@@ -75,18 +75,19 @@
 parts = dolphin.split()
 i = 0
 code_map = {
-    'M': (Path.MOVETO, 1),
-    'C': (Path.CURVE4, 3),
-    'L': (Path.LINETO, 1)}
+    'M': Path.MOVETO,
+    'C': Path.CURVE4,
+    'L': Path.LINETO,
+}
 
 while i < len(parts):
-    code = parts[i]
-    path_code, npoints = code_map[code]
+    path_code = code_map[parts[i]]
+    npoints = Path.NUM_VERTICES_FOR_CODE[path_code]
     codes.extend([path_code] * npoints)
-    vertices.extend([[float(x) for x in y.split(',')] for y in
-                     parts[i + 1:i + npoints + 1]])
+    vertices.extend([[*map(float, y.split(','))]
+                     for y in parts[i + 1:][:npoints]])
     i += npoints + 1
-vertices = np.array(vertices, float)
+vertices = np.array(vertices)
 vertices[:, 1] -= 160
 
 dolphin_path = Path(vertices, codes)
