Cast to np.array earlier (in set_3d_properties)

krassowski · krassowski · commit f88d1f6e559a · 2022-07-18T00:34:12.000+01:00
diff --git a/lib/mpl_toolkits/mplot3d/art3d.py b/lib/mpl_toolkits/mplot3d/art3d.py
@@ -545,7 +545,7 @@ def set_3d_properties(self, zs, zdir):
         #
         # Grab the current sizes and linewidths to preserve them.
         self._sizes3d = self._sizes
-        self._linewidths3d = self._linewidths
+        self._linewidths3d = np.array(self._linewidths)
         xs, ys, zs = self._offsets3d
 
         # Sort the points based on z coordinates
@@ -600,7 +600,7 @@ def do_3d_projection(self):
             self._sizes = self._sizes3d[z_markers_idx]
 
         if len(self._linewidths3d) > 1:
-            self._linewidths = np.array(self._linewidths3d)[z_markers_idx]
+            self._linewidths = self._linewidths3d[z_markers_idx]
 
         # Re-order items
         vzs = vzs[z_markers_idx]
