OwlCyberSecurity - MANAGER
Edit File: stackplot.cpython-38.pyc
U ��]p � @ s( d Z ddlZdgZdddd�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 O s^ t �|�}t|�}|dk r&| j|d� t j|dt �|jt j�d�}|dkrRd} �nR|dkr�t �|d� d } || ddd�f 7 }�n |d kr�|j d } || d t � | �dd�df �d�} | | } || 7 }n�|d k�r�t �|d�} t � | �} | dk} d | | | | < t � |dd�dd �f t � |�f�}| | }|d| 7 }|| }d|dd�df< |d | }t �|�d��}|d| } || 7 }nd | }|d7 }t|��| j�� }| j|| |ddd�f f|t|d�d�|��}dg|jjdd�< |g}tt|�d �D ]T}| j�� }|�| j|||dd�f ||d dd�f f|t|d�d�|��� �q|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�dtyper g Zsymg �?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�xr r r �args�kwargsr �stackZ first_line�mZtotalZ inv_total�maskZincreaseZ below_sizeZmove_up�centerZerrstrr Zcoll�r�ir r �6/usr/lib/python3/dist-packages/matplotlib/stackplot.pyr sd / ( $ �� , �� )�__doc__Znumpyr �__all__r r r r r �<module> s �