Fix undefined variables

rth · rth · commit a3c331dd98e6 · 2018-07-05T15:25:26.000+02:00
diff --git a/lib/mpl_toolkits/mplot3d/art3d.py b/lib/mpl_toolkits/mplot3d/art3d.py
@@ -141,7 +141,7 @@ def set_3d_properties(self, zs=0, zdir='z'):
         xs = self.get_xdata()
         ys = self.get_ydata()
 
-        if not iterable(zs):
+        if not cbook.iterable(zs):
             zs = np.ones(len(xs)) * zs
         xyz = np.asarray([xs, ys, zs])
         self._verts3d = juggle_axes_vec(xyz, zdir)
@@ -181,7 +181,7 @@ def path_to_3d_segment(path, zs=0, zdir='z'):
 def paths_to_3d_segments(paths, zs=0, zdir='z'):
     """Convert paths from a collection object to 3D segments."""
 
-    if not iterable(zs):
+    if not cbook.iterable(zs):
         zs = np.ones(len(paths)) * zs
 
     segments = []
@@ -401,7 +401,7 @@ def set_3d_properties(self, zs, zdir):
         # Force the collection to initialize the face and edgecolors
         # just in case it is a scalarmappable with a colormap.
         self.update_scalarmappable()
-        offsets = np.vstack(self.get_offsets(), np.ones(len(verts)) * zs)
+        offsets = np.vstack(self.get_offsets(), np.atleast_1d(zs))
         self._offsets3d = juggle_axes_vec(offsets, zdir)
         self._facecolor3d = self.get_facecolor()
         self._edgecolor3d = self.get_edgecolor()
