B
    ·ôJd0  ã               @   s4   d Z ddlZddlmZ dgZddddœdd„ZdS )	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         
   O   s^  t  |¡}t|ƒ}|dk	r&| j|d t j|dt  |jt j¡d}tj	ddddg|d	 |dkrhd
}	n8|dkršt  
|d¡ d }	||	ddd…f 7 }n|dkrè|jd }
||
d t  |
¡dd…df    
d¡}	|	|
  }	||	7 }n¸|dkr t  
|d¡}t  |¡}|dk}d||  ||< t  |dd…dd…f t  |¡f¡}|| }|d| 7 }|| }d|dd…df< |d | }t  | 
d¡¡}|d|  }	||	7 }| j ¡ }| j||	|ddd…f f|t|dƒdœ|—Ž}dg|jjdd…< |g}xhtt|ƒd ƒD ]T}| j ¡ }| | j|||dd…f ||d dd…f f|t|dƒdœ|—Ž¡ qW |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   )ZaxisÚdtyper   ÚsymZwiggleZweighted_wiggle)r   g        g      à?g      ð?é   )Z	facecolorÚlabel)ÚnpZ	row_stackÚiterZset_prop_cycleZcumsumZpromote_typesr
   Zfloat32r   Zcheck_in_listÚsumÚshapeZarangeZ
zeros_likeZhstackÚdiffZ
_get_linesZget_next_colorZfill_betweenÚnextZsticky_edgesÚyÚrangeÚlenÚappend)ZaxesÚxr   r   r   ÚargsÚkwargsr   ÚstackÚ
first_lineÚmÚtotalZ	inv_totalÚmaskZincreaseZ
below_sizeZmove_upÚcenterr	   ZcollÚrÚir   r   úF/var/www/html/venv/lib/python3.7/site-packages/matplotlib/stackplot.pyr      sV    5

(



$

,)Ú__doc__Únumpyr   Z
matplotlibr   Ú__all__r   r   r   r   r#   Ú<module>   s   