Replace an unnecessary zip() in mplot3d by numpy ops.

anntzer · anntzer · commit a3ed0fdfea83 · 2018-06-28T14:14:05.000+02:00
Collection.set_offsets ensures that `offsets` is a numpy array, so at
least in the case where the array is not empty, `xs` and `ys` can just
be views.

Probably a tiny bit faster.
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 = []
