
    G8c]                     :    d Z ddlZddlZddlmZ dgZdddddZy)	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                V    t        j                  |      }t        |      }|t        j                  |      }n fd|D        }t        j
                  |dt        j                  |j                  t         j                              }t        j                  g d|       |dk(  rd}	nH|d	k(  r(t        j                  |d       d
z  }	||	dddf   z  }n|dk(  rO|j                  d   }
||
d
z
  t        j                  |
      dddf   z
  z  j                  d      }	|	|
 z  }	||	z  }n|dk(  rt        j                  |d      }t        j                  |      }|dkD  }d||   z  ||<   t        j                  |ddddf   t        j                   |      f      }||z
  }|d
|z  z  }||z  }d
|dddf<   |d
z
  |z  }t        j
                  |j                  d            }|d
|z  z
  }	||	z  }  j"                  |	|dddf   ft%        |      t%        |d      d|}dg|j&                  j(                  dd |g}t+        t-        |      dz
        D ]M  }|j/                    j"                  |||ddf   ||dz   ddf   ft%        |      t%        |d      d|       O |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.
    Nc              3   R   K   | ]  }j                   j                            y w)N)
_get_linesget_next_color).0_axess     6/usr/lib/python3/dist-packages/matplotlib/stackplot.py	<genexpr>zstackplot.<locals>.<genexpr>M   s     >q$//002>s   $'r   )axisdtype)r   symwiggleweighted_wiggle)r	   r   g        r   g      ?r   r   g      ?   )	facecolorlabel)np	row_stackiter	itertoolscyclecumsumpromote_typesr   float32r   check_in_listsumshapearange
zeros_likehstackdifffill_betweennextsticky_edgesyrangelenappend)r   xr   r   r	   argskwargsr-   stack
first_linemtotal	inv_totalmaskincrease
below_sizemove_upcentercollris   `                    r   r   r      s   j 	TA&\F(>A> IIaar'7'7'LMEC (*6
	U	ffQl]S(
D!G$$	X	GGAJ1s7RYYq\!T'%::;@@C
qb
	&	&q!MM%(	qyd+	$99a1Q3i45U]
cAg
y(1C-8+6::a=)cEk)
 4Q
E!Q$K ''+F|4;M'%'D  SD	A 3q6A: .	"""1eAqDk5Q? --1&\)-fd);- &,- 	..
 H    )__doc__r   numpyr   
matplotlibr   __all__r   r   rA   r   <module>rF      s+      - vmrA   