Merge pull request #11528 from anntzer/zipless

tacaswell · web-flow · commit d896d6ddbfd8 · 2018-06-28T12:11:24.000-04:00
MNT: Replace an unnecessary zip() in mplot3d by numpy ops.
diff --git a/lib/mpl_toolkits/mplot3d/art3d.py b/lib/mpl_toolkits/mplot3d/art3d.py
@@ -380,7 +380,7 @@ def set_3d_properties(self, zs, zdir):
         self.update_scalarmappable()
         offsets = self.get_offsets()
         if len(offsets) > 0:
-            xs, ys = zip(*offsets)
+            xs, ys = offsets.T
         else:
             xs = []
             ys = []
@@ -446,7 +446,7 @@ def set_3d_properties(self, zs, zdir):
         self.update_scalarmappable()
         offsets = self.get_offsets()
         if len(offsets) > 0:
-            xs, ys = zip(*offsets)
+            xs, ys = offsets.T
         else:
             xs = []
             ys = []
