B ·ôJd0ã@s4dZddlZddlmZdgZddddœdd„ZdS) 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)Ú_apiÚ stackplot©Úzero)ÚlabelsÚcolorsÚbaselinec Os^t |¡}t|ƒ}|dk r&|j|dtj|dt |jtj¡d}tj ddddg|d |dkrhd } n8|dkršt  |d¡ d } || ddd…f7}n|dkrè|j d} || d t  | ¡dd…df  d¡} | | } || 7}n¸|dkr t  |d¡} t  | ¡} | dk} d | | | | <t |dd…dd …ft |¡f¡}| |}|d |7}|| }d |dd…df<|d |}t |  d¡¡}|d | } || 7}|j ¡}|j|| |ddd…ff|t|dƒdœ|—Ž}dg|jjdd…<|g}xhtt|ƒd ƒD]T}|j ¡}| |j|||dd…f||d dd…ff|t|dƒdœ|—Ž¡qW|S)aÚ Draw a stacked area plot. Parameters ---------- x : (N,) array-like y : (M, N) array-like The data is assumed to be unstacked. Each of the following calls is legal:: stackplot(x, y) # where y has shape (M, N) stackplot(x, y1, y2, y3) # 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 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`. Returns ------- list of `.PolyCollection` A list of `.PolyCollection` instances, one for each element in the stacked area plot. N)Úcolorr)ZaxisÚdtyperÚsymZwiggleZweighted_wiggle)rggà?gð?é)Z facecolorÚlabel)ÚnpZ row_stackÚiterZset_prop_cycleZcumsumZ promote_typesr Zfloat32rZ check_in_listÚsumÚshapeZarangeZ zeros_likeZhstackÚdiffZ _get_linesZget_next_colorZ fill_betweenÚnextZ sticky_edgesÚyÚrangeÚlenÚappend)ZaxesÚxrrrÚargsÚkwargsrÚstackÚ first_lineÚmÚtotalZ inv_totalÚmaskZincreaseZ below_sizeZmove_upÚcenterr ZcollÚrÚirrúF/var/www/html/venv/lib/python3.7/site-packages/matplotlib/stackplot.pyrsV5   (     $     ,)Ú__doc__ÚnumpyrZ matplotlibrÚ__all__rrrrr#Ús