Fix array-like linewidth for 3d scatter

Author: krassowski

lib/mpl_toolkits/mplot3d/art3d.py

@@ -600,7 +600,7 @@ def do_3d_projection(self):
             self._sizes = self._sizes3d[z_markers_idx]
 
         if len(self._linewidths3d) > 1:
-            self._linewidths = self._linewidths3d[z_markers_idx]
+            self._linewidths = np.array(self._linewidths3d)[z_markers_idx]
 
         # Re-order items
         vzs = vzs[z_markers_idx]

lib/mpl_toolkits/tests/baseline_images/test_mplot3d/scatter3d_linewidth.png

@@ -299,6 +299,16 @@ def test_scatter3d_color():
                color='b', marker='s')
 
 
+@mpl3d_image_comparison(['scatter3d_linewidth.png'])
+def test_scatter3d_linewidth():
+    fig = plt.figure()
+    ax = fig.add_subplot(projection='3d')
+
+    # Check that array-like linewidth can be set
+    ax.scatter(np.arange(10), np.arange(10), np.arange(10),
+               marker='o', linewidth=np.arange(10))
+
+
 @check_figures_equal(extensions=['png'])
 def test_scatter3d_modification(fig_ref, fig_test):
     # Changing Path3DCollection properties post-creation should work correctly.