FilesExpand file tree

 master

/

stackplot.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

60 lines (43 loc) · 1.8 KB

 master

/

stackplot.py

Copy path

Top

File metadata and controls

• Code
• Blame

60 lines (43 loc) · 1.8 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

"""

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)

"""

import numpy as np

__all__ = ['stackplot']

def stackplot(axes, x, *args, **kwargs):

"""Draws a stacked area plot.

*x* : 1d array of dimension N

*y* : 2d array of dimension MxN, OR any number 1d arrays each of 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

Keyword arguments:

*colors* : A list or tuple of colors. These will be cycled through and

used to colour the stacked areas.

All other keyword arguments are passed to

:func:`~matplotlib.Axes.fill_between`

Returns *r* : A list of

:class:`~matplotlib.collections.PolyCollection`, one for each

element in the stacked area plot.

"""

if len(args) == 1:

y = np.atleast_2d(*args)

elif len(args) > 1:

y = np.row_stack(args)

colors = kwargs.pop('colors', None)

if colors is not None:

axes.set_color_cycle(colors)

# Assume data passed has not been 'stacked', so stack it here.

y_stack = np.cumsum(y, axis=0)

r = []

# Color between x = 0 and the first array.

r.append(axes.fill_between(x, 0, y_stack[0, :],

facecolor=axes._get_lines.color_cycle.next(), **kwargs))

# Color between array i-1 and array i

for i in xrange(len(y) - 1):

r.append(axes.fill_between(x, y_stack[i, :], y_stack[i + 1, :],

facecolor=axes._get_lines.color_cycle.next(), **kwargs))

return r