3 ƽhD@s2dZddlZddljZdgZfdddddZdS)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 stackplotzero)labelscolorsbaselinec Os^tj|}t|}|dk r&|j|dtj|dtj|jtjd}tj ddddg|d |dkrhd } n8|dkrtj |d d } || dddf7}n|dkr|j d} || d tj | dddfj d} | | } || 7}n|dkrtj |d} tj | } | dk} d | | | | <tj|dddd ftj|f}| |}|d |7}|| }d |dddf<|d |}tj|j d}|d | } || 7}|jj}|j|| |dddff|t|dd|}dg|jjdd<|g}xhtt|d D]T}|jj}|j|j|||ddf||d ddff|t|dd|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 of `.PolyCollection` A list of `.PolyCollection` instances, one for each element in the stacked area plot. N)colorr)ZaxisdtypersymZwiggleZweighted_wiggle)rgg?g?)Z facecolorlabel)npZ row_stackiterZset_prop_cycleZcumsumZ promote_typesrZfloat32cbookZ_check_in_listsumshapeZarangeZ zeros_likeZhstackZdiffZ _get_linesZget_next_colorZ fill_betweennextZ sticky_edgesyrangelenappend)Zaxesxrrrargskwargsrstack first_linemtotalZ inv_totalmaskZincreaseZ below_sizeZmove_upcenterrZcollrir!:/tmp/pip-build-7iwl8md4/matplotlib/matplotlib/stackplot.pyrsV.   (     $     ,)__doc__numpyr Zmatplotlib.cbookr__all__rr!r!r!r"s