Update lib/mpl_toolkits/mplot3d/axes3d.py

jgehrcke · jgehrcke · commit 29e556a63e70 · 2012-09-23T21:14:40.000+03:00
Make Axes3D.scatter properly deal with `zs` being of size 1.
diff --git a/lib/mpl_toolkits/mplot3d/axes3d.py b/lib/mpl_toolkits/mplot3d/axes3d.py
@@ -1939,7 +1939,7 @@ def scatter(self, xs, ys, zs=0, zdir='z', s=20, c='b', *args, **kwargs):
             raise ValueError("Arguments 'xs' and 'ys' must be of same size.")
         if xs.size != zs.size:
             if zs.size == 1:
-                zs = np.array(zs[0] * xs.size)
+                zs = np.tile(zs[0], xs.size)
             else:
                 raise ValueError(("Argument 'zs' must be of same size as 'xs' "
                     "and 'ys' or of size 1."))
