Check if zs has same length as paths, raise else

thangleiter · web-flow · commit f24663fb3091 · 2020-08-08T19:03:31.000+02:00
diff --git a/lib/mpl_toolkits/mplot3d/art3d.py b/lib/mpl_toolkits/mplot3d/art3d.py
@@ -209,6 +209,9 @@ def _paths_to_3d_segments(paths, zs=0, zdir='z'):
 
     if not np.iterable(zs):
         zs = np.broadcast_to(zs, len(paths))
+    else:
+        if len(zs) != len(paths):
+            raise ValueError('Number of z-coordinates does not match paths.')
 
     segs = [_path_to_3d_segment(path, pathz, zdir)
             for path, pathz in zip(paths, zs)]
