Slightly improve streamplot code legibility.

anntzer · anntzer · commit 1820339c3498 · 2020-02-06T10:20:30.000+01:00
`__getitem__` is never called with more than a single arg, and we can
just use normal indexing on _mask both for getting and for setting.
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.
