Merge pull request #16423 from anntzer/streamplot

dstansby · web-flow · commit bd02c9d1147f · 2020-02-06T21:03:48.000Z
Slightly improve streamplot code legibility.
diff --git a/lib/matplotlib/streamplot.py b/lib/matplotlib/streamplot.py
@@ -380,8 +380,8 @@ def __init__(self, density):
 
         self._current_xy = None
 
-    def __getitem__(self, *args):
-        return self._mask.__getitem__(*args)
+    def __getitem__(self, args):
+        return self._mask[args]
 
     def _start_trajectory(self, xm, ym):
         """Start recording streamline trajectory"""
@@ -391,7 +391,7 @@ def _start_trajectory(self, xm, ym):
     def _undo_trajectory(self):
         """Remove current trajectory from mask"""
         for t in self._traj:
-            self._mask.__setitem__(t, 0)
+            self._mask[t] = 0
 
     def _update_trajectory(self, xm, ym):
         """Update current trajectory position in mask.
