Merge pull request matplotlib#2172 from WeatherGod/mplot3d/iter_collections

pelson · pelson · commit 11666e52dc34 · 2013-06-29T09:20:11.000-07:00
[SPRINT] Closes matplotlib#1079. Allow iterables for Line3DCollection
diff --git a/lib/mpl_toolkits/mplot3d/art3d.py b/lib/mpl_toolkits/mplot3d/art3d.py
@@ -106,9 +106,11 @@ def set_3d_properties(self, zs=0, zdir='z'):
         ys = self.get_ydata()
 
         try:
+            # If *zs* is a list or array, then this will fail and
+            # just proceed to juggle_axes().
             zs = float(zs)
             zs = [zs for x in xs]
-        except:
+        except TypeError:
             pass
         self._verts3d = juggle_axes(xs, ys, zs, zdir)
 
@@ -170,7 +172,7 @@ def set_segments(self, segments):
         '''
         Set 3D segments
         '''
-        self._segments3d = segments
+        self._segments3d = np.asanyarray(segments)
         LineCollection.set_segments(self, [])
 
     def do_3d_projection(self, renderer):
