matplotlib.github.com/_modules/matplotlib/stackplot.html at master · 2236zml/matplotlib.github.com

History

261 lines (209 loc) · 19.8 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

211

212

213

214

215

216

217

218

219

220

221

222

223

224

225

226

227

228

229

230

231

232

233

234

235

236

237

238

239

240

241

242

243

244

245

246

247

248

249

250

251

252

253

254

255

256

257

258

259

260

261

<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN"

"http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">

<head>

<title>matplotlib.stackplot — Matplotlib 3.1.3 documentation</title>

type="text/css" />

type="text/css" />

title="Search within Matplotlib 3.1.3 documentation"

href="../../_static/opensearch.xml"/>

</head>

<body>

<div style="float: left; position: absolute; width: 496px; bottom: 0; padding-bottom: 24px">Version 3.1.3</div>

<div id="forkongithub"><a href="https://github.com/matplotlib/matplotlib">Fork me on GitHub</a></div>

</div>

<ul>

<li><a href="../../users/installing.html">Installation</a></li>

<li><a href="../../contents.html">Documentation</a></li>

<li><a href="../../gallery/index.html">Examples</a></li>

<li><a href="../../tutorials/index.html">Tutorials</a></li>

<li><a href="../../devel/index.html">Contributing</a></li>

</form>

</li>

</ul>

</nav>

<h3>Navigation</h3>

<ul>

<a href="../../genindex.html" title="General Index"

accesskey="I">index</a></li>

<a href="../../py-modindex.html" title="Python Module Index"

>modules</a> |</li>

<li><a href="../../contents.html">contents</a> »</li>

<li><a href="../index.html" >Module code</a> »</li>

<li><a href="../matplotlib.html" accesskey="U">matplotlib</a> »</li>

</ul>

</div>

<h3>Related Topics</h3>

<ul>

<li><a href="../../contents.html">Documentation overview</a><ul>

<li><a href="../index.html">Module code</a><ul>

<li><a href="../matplotlib.html">matplotlib</a><ul>

</ul></li>

</ul>

</div>

<h1>Source code for matplotlib.stackplot</h1><div class="highlight"><pre>

"""

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,

labels=(), colors=None, baseline='zero',

**kwargs):

"""

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.

"""

y = np.row_stack(args)

labels = iter(labels)

if colors is not None:

axes.set_prop_cycle(color=colors)

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

# We'll need a float buffer for the upcoming calculations.

stack = np.cumsum(y, axis=0, dtype=np.promote_types(y.dtype, np.float32))

if baseline == 'zero':

first_line = 0.

elif baseline == 'sym':

first_line = -np.sum(y, 0) * 0.5

stack += first_line[None, :]

elif baseline == 'wiggle':

m = y.shape[0]

first_line = (y * (m - 0.5 - np.arange(m)[:, None])).sum(0)

first_line /= -m

stack += first_line

elif baseline == 'weighted_wiggle':

total = np.sum(y, 0)

# multiply by 1/total (or zero) to avoid infinities in the division:

inv_total = np.zeros_like(total)

mask = total > 0

inv_total[mask] = 1.0 / total[mask]

increase = np.hstack((y[:, 0:1], np.diff(y)))

below_size = total - stack

below_size += 0.5 * y

move_up = below_size * inv_total

move_up[:, 0] = 0.5

center = (move_up - 0.5) * increase

center = np.cumsum(center.sum(0))

first_line = center - 0.5 * total

stack += first_line

else:

errstr = "Baseline method %s not recognised. " % baseline

errstr += "Expected 'zero', 'sym', 'wiggle' or 'weighted_wiggle'"

raise ValueError(errstr)

# Color between x = 0 and the first array.

color = axes._get_lines.get_next_color()

coll = axes.fill_between(x, first_line, stack[0, :],

facecolor=color, label=next(labels, None),

**kwargs)

coll.sticky_edges.y[:] = [0]

r = [coll]

# Color between array i-1 and array i

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

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

**kwargs))

return r

</pre></div>

</div>

</div>

Last updated on Feb 09, 2020.

Created using

<a href="http://sphinx-doc.org/">Sphinx</a> 1.8.5.

Doc version v3.1.3-2-g782f6ef56.

</div>

(function(i,s,o,g,r,a,m){i['GoogleAnalyticsObject']=r;i[r]=i[r]||function(){

(i[r].q=i[r].q||[]).push(arguments)},i[r].l=1*new Date();a=s.createElement(o),

m=s.getElementsByTagName(o)[0];a.async=1;a.src=g;m.parentNode.insertBefore(a,m)

})(window,document,'script','//www.google-analytics.com/analytics.js','ga');

ga('create', 'UA-55954603-1', 'auto');

ga('send', 'pageview');

</script>

</body>

</footer>

</html>

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

FilesExpand file tree

stackplot.html

Latest commit

History

stackplot.html

File metadata and controls