streamcontainer

story645 · story645 · commit 25ba160dc6e0 · 2022-11-06T22:59:09.000-05:00
diff --git a/lib/matplotlib/streamplot.py b/lib/matplotlib/streamplot.py
@@ -7,6 +7,7 @@
 
 import matplotlib as mpl
 from matplotlib import _api, cm, patches
+from matplotlib.container import Container
 import matplotlib.colors as mcolors
 import matplotlib.collections as mcollections
 import matplotlib.lines as mlines
@@ -76,17 +77,7 @@ def streamplot(axes, x, y, u, v, density=1, linewidth=None, color=None,
 
     Returns
     -------
-    StreamplotSet
-        Container object with attributes
-
-        - ``lines``: `.LineCollection` of streamlines
-
-        - ``arrows``: `.PatchCollection` containing `.FancyArrowPatch`
-          objects representing the arrows half-way along stream lines.
-
-        This container will probably change in the future to allow changes
-        to the colormap, alpha, etc. for both lines and arrows, but these
-        changes should be backward compatible.
+    `~.StreamplotContainer`
     """
     grid = Grid(x, y)
     mask = StreamMask(density)
@@ -237,20 +228,50 @@ def streamplot(axes, x, y, u, v, density=1, linewidth=None, color=None,
         axes.add_patch(p)
 
     axes.autoscale_view()
-    stream_container = StreamplotSet(lc, ac)
+    stream_container = StreamplotContainer((lc, ac))
     return stream_container
 
 
+@_api.deprecated("major.minor", alternative="streamplot.StreamplotContainer")
 class StreamplotSet:
-
     def __init__(self, lines, arrows):
         self.lines = lines
         self.arrows = arrows
 
 
+class StreamplotContainer(Container):
+    """
+    `~.Container` object for artists created in an `~.Axes.streamplot` plot.
+
+    It can be treated like a namedtuple with fields ``(lines, arrows)``
+
+    .. versionadded major.minor ::
+
+    Attributes
+    ----------
+    lines_arrows : tuple
+        Tuple of (lines, arrows)
+        ``lines``: `.LineCollection` of the streamlines.
+        ``arrows``: `.PatchCollection` of the `.FancyArrowPatch` arrows
+                    half-way along each streamlines.
+
+    .. note::
+        This container will probably change in the future to allow changes
+        to the colormap, alpha, etc. for both lines and arrows, but these
+        changes should be backward compatible.
+    """
+
+    def __init__(self, lines_arrows, **kwargs):
+        lines, arrows = lines_arrows
+        self.lines = lines
+        self.arrows = arrows
+        super().__init__(lines_arrows, **kwargs)
+
+
 # Coordinate definitions
 # ========================
 
+
 class DomainMap:
     """
     Map representing different coordinate systems.
