B ]”t\‰ã@s(dZddlZdgZddddœdd„ZdS)zâ Stacked area plot for 1D arrays inspired by Douglas Y'barbo's stackoverflow answer: http://stackoverflow.com/questions/2225995/how-can-i-create-stacked-line-graph-with-matplotlib (http://stackoverflow.com/users/66549/doug) éNÚ stackplot©Úzero)ÚlabelsÚcolorsÚbaselinec Oslt |¡}t|ƒ}|dk r&|j|dtj|dt |jtj¡d}|dkrRd} n\|dkr„t |d¡ d} || ddd…f7}n*|d krÒ|j d} || dt  | ¡dd…df d¡} | | } || 7}nÜ|d kr–|j \} } t |d¡} t  | ¡} | dk}d | || |<t  |dd…dd …ft  |¡f¡}| |}|d|7}|| }d|dd…df<|d|}t | d¡¡}|d| } || 7}nd |}|d7}t|ƒ‚|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 : 1d array of dimension N y : 2d array (dimension MxN), or sequence of 1d arrays (each dimension 1xN) The data is assumed to be unstacked. Each of the following calls is legal:: stackplot(x, y) # where y is MxN stackplot(x, y1, y2, y3, y4) # where y1, y2, y3, y4, are all 1xNm 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 : Length N sequence of strings Labels to assign to each data series. colors : Length N sequence of colors A list or tuple of colors. These will be cycled through and used to colour the stacked areas. **kwargs : All other keyword arguments are passed to `Axes.fill_between()`. Returns ------- list : list of `.PolyCollection` A list of `.PolyCollection` instances, one for each element in the stacked area plot. N)Úcolorr)ZaxisÚdtypergZsymgà?ZwiggleZweighted_wigglegð?éz#Baseline method %s not recognised. z5Expected 'zero', 'sym', 'wiggle' or 'weighted_wiggle')Z facecolorZlabel)ÚnpZ row_stackÚiterZset_prop_cycleZcumsumZ promote_typesr Zfloat32ÚsumÚshapeZarangeZ zeros_likeZhstackZdiffÚ ValueErrorZ _get_linesZget_next_colorZ fill_betweenÚnextZ sticky_edgesÚyÚrangeÚlenÚappend)ZaxesÚxrrrÚargsÚkwargsrÚstackZ first_lineÚmÚnZtotalZ inv_totalÚmaskZincreaseZ below_sizeZmove_upÚcenterZerrstrrZcollÚrÚirrú3lib/python3.7/site-packages/matplotlib/stackplot.pyrsZ/   (      $      ,)Ú__doc__Znumpyr Ú__all__rrrrrÚs