Finally, mask linewidths

dstansby · dstansby · commit 7cabe139ddee · 2016-12-06T14:30:30.000Z
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -3995,10 +3995,11 @@ def scatter(self, x, y, s=None, c=None, marker=None, cmap=None, norm=None,
         else:
             colors = None  # use cmap, norm after collection is created
 
-        # s, c, colors, or edgecolors will be unchanged unless they are the
-        # same length as x:
-        maskargs = x, y, s, c, colors, edgecolors
-        x, y, s, c, colors, edgecolors = cbook.delete_masked_points(*maskargs)
+        # Anything in maskargs will be unchanged unless it is the same length
+        # as x:
+        maskargs = x, y, s, c, colors, edgecolors, linewidths
+        x, y, s, c, colors, edgecolors, linewidths =\
+            cbook.delete_masked_points(*maskargs)
 
         scales = s   # Renamed for readability below.
 
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -4848,9 +4848,12 @@ def test_scatter_color_masking():
     x = np.array([1, 2, 3])
     y = np.array([1, np.nan, 3])
     colors = np.array(['k', 'w', 'k'])
+    linewidths = np.array([1, 2, 3])
     s = plt.scatter(x, y, color=colors)
 
     facecolors = s.get_facecolors()
     linecolors = s.get_edgecolors()
+    linewidths = s.get_linewidths()
     assert_array_equal(facecolors[1], np.array([0, 0, 0, 1]))
     assert_array_equal(linecolors[1], np.array([0, 0, 0, 1]))
+    assert linewidths[1] == 3
