l0j@dZddlZddlmZmZmZmZdgZdddddZ dS) z Stacked area plot for 1D arrays inspired by Douglas Y'barbo's stackoverflow answer: https://stackoverflow.com/q/2225995/ (https://stackoverflow.com/users/66549/doug) N)cbook collections_api_style_helpers stackplotzero)labelscolorsbaselinec tj|}t|}|fd|D}tj|t j}|d|tj |\}}tj |dtj |j tj } tjgd||dkrd } nW|d kr*tj|d d z} | | dddfz } n'|d krS|jd} || d z tj| dddfz zd} | | z} | | z } n|d krtj|d} tj| } | dk}d| |z | |<tj|ddddftj|f}| | z }|d |zz }|| z}d |dddf<|d z |z}tj |d}|d | zz } | | z } j|| | dddffdt/|dit/||}dg|jjdd<|g}t5t7|dz D][}|j|| |ddf| |dzddffdt/|dit/||\|S)a Draw a stacked area plot or a streamgraph. Parameters ---------- x : (N,) array-like y : (M, N) array-like The data can be either stacked or unstacked. Each of the following calls is legal:: stackplot(x, y) # where y has shape (M, N) e.g. y = [y1, y2, y3, y4] stackplot(x, y1, y2, y3, y4) # where y1, y2, y3, y4 have length N baseline : {'zero', 'sym', 'wiggle', 'weighted_wiggle'} Method used to calculate the baseline: - ``'zero'``: Constant zero baseline, i.e. a simple stacked plot. - ``'sym'``: Symmetric around zero and is sometimes called 'ThemeRiver'. - ``'wiggle'``: Minimizes the sum of the squared slopes. - ``'weighted_wiggle'``: Does the same but weights to account for size of each layer. It is also called 'Streamgraph'-layout. More details can be found at http://leebyron.com/streamgraph/. labels : list of str, optional A sequence of labels to assign to each data series. If unspecified, then no labels will be applied to artists. colors : list of :mpltype:`color`, optional A sequence of colors to be cycled through and used to color the stacked areas. The sequence need not be exactly the same length as the number of provided *y*, in which case the colors will repeat from the beginning. If not specified, the colors from the Axes property cycle will be used. data : indexable object, optional DATA_PARAMETER_PLACEHOLDER **kwargs All other keyword arguments are passed to `.Axes.fill_between`. The following parameters additionally accept a sequence of values corresponding to the *y* datasets: - *hatch* - *edgecolor* - *facecolor* - *linewidth* - *linestyle* .. versionadded:: 3.9 Allowing a sequence of strings for *hatch*. .. versionadded:: 3.11 Allowing sequences of values in above listed `.Axes.fill_between` parameters. Returns ------- list of `.PolyCollection` A list of `.PolyCollection` instances, one for each element in the stacked area plot. NcBg|]}jSr) _get_linesget_next_color).0_axess W/home/wildlama/miniconda3/envs/lam/lib/python3.11/site-packages/matplotlib/stackplot.py zstackplot..Ys'>>>q$/0022>>> facecolorr)axisdtype)r symwiggleweighted_wiggle)r r grg?rrg?label)npvstackiterrnormalize_kwargsrPolyCollection setdefaultrstyle_generatorcumsum promote_typesrfloat32r check_in_listsumshapearange zeros_likehstackdiff fill_betweennext sticky_edgesyrangelenappend)rxr r r argskwargsr3 style_genstack first_linemtotal inv_totalmaskincrease below_sizemove_upcentercollris` rrrsH $A &\\F ~>>>>A>>>  #FK,F G GF k6***&6v>>FI Iaar'7'L'L M M MECCC (****6 U  fQll]S(  D!!!G$$ X   GAJ1s7RYq\\!!!T'%::;@@CC qb   & & &q! M%(( qyd + $9a1Q3i455U] cAg y(1 C-8+6::a==))cEk)   4 Q E!QQQ$K : :#'#5#5 :#I :28 : :D SD A3q66A:  AA ""1eAqqqDk5Q?@@)-fd););@%))__@8>@@ A A A A Hr) __doc__numpyr matplotlibrrrr__all__rrrrrLs{???????????? -v~ ~ ~ ~ ~ ~ ~ r