matplotlib.github.com/2.2.0/gallery/animation/rain.html at main · matplotlib/matplotlib.github.com · GitHub

History

225 lines (188 loc) · 19.4 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

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

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

<head>

<title>Rain simulation — Matplotlib 2.2.0 documentation</title>

type="text/css" />

type="text/css" />

var DOCUMENTATION_OPTIONS = {

URL_ROOT: '../../',

VERSION: '2.2.0',

COLLAPSE_INDEX: false,

FILE_SUFFIX: '.html',

HAS_SOURCE: true,

SOURCELINK_SUFFIX: '.txt'

};

</script>

title="Search within Matplotlib 2.2.0 documentation"

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

</head>

<body>

<div id="unreleased-message"> You are reading an old version of the documentation (v2.2.0). For the latest version see <a href="https://matplotlib.org/stable/gallery/animation/rain.html">https://matplotlib.org/stable/gallery/animation/rain.html</a></div>

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

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

</div>

<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>

<a href="random_walk.html" title="Animated 3D random walk"

accesskey="N">next</a> |</li>

<a href="animated_histogram.html" title="Animated histogram"

accesskey="P">previous</a> |</li>

<li><a href="../index.html">examples</a>| </li>

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

<li><a href="../../api/pyplot_summary.html">pyplot</a>| </li>

<li><a href="../../api/index.html" >The Matplotlib API</a> »</li>

<li><a href="../../api/animation_api.html" accesskey="U">animation</a> »</li>

</ul>

</div>

<h3>Related Topics</h3>

<ul>

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

<li><a href="../../api/index.html">The Matplotlib API</a><ul>

<li><a href="../../api/animation_api.html">animation</a><ul>

<li>Previous: <a href="animated_histogram.html" title="previous chapter">Animated histogram</a></li>

<li>Next: <a href="random_walk.html" title="next chapter">Animated 3D random walk</a></li>

</ul></li>

</ul>

</div>

<li><a href="../../_sources/gallery/animation/rain.rst.txt"

rel="nofollow">Show Source</a></li>

</ul>

</div>

<h3>Quick search</h3>

</form>

</div>

</div>

<h1>Rain simulation<a class="headerlink" href="#rain-simulation" title="Permalink to this headline">¶</a></h1>

Simulates rain drops on a surface by animating the scale and opacity

of 50 scatter points.

Author: Nicolas P. Rougier

<div class="highlight-python notranslate"><div class="highlight"><pre>import numpy as np

import matplotlib.pyplot as plt

from matplotlib.animation import <a href="../../api/_as_gen/matplotlib.animation.FuncAnimation.html#matplotlib.animation.FuncAnimation" title="View documentation for matplotlib.animation.FuncAnimation">FuncAnimation</a>

# Fixing random state for reproducibility

<a href="https://docs.scipy.org/doc/numpy/reference/generated/numpy.random.seed.html#numpy.random.seed" title="View documentation for numpy.random.seed">np.random.seed</a>(19680801)

# Create new Figure and an Axes which fills it.

fig = <a href="../../api/_as_gen/matplotlib.pyplot.figure.html#matplotlib.pyplot.figure" title="View documentation for matplotlib.pyplot.figure">plt.figure</a>(figsize=(7, 7))

ax = fig.add_axes([0, 0, 1, 1], frameon=False)

ax.set_xlim(0, 1), ax.set_xticks([])

ax.set_ylim(0, 1), ax.set_yticks([])

# Create rain data

n_drops = 50

rain_drops = <a href="https://docs.scipy.org/doc/numpy/reference/generated/numpy.zeros.html#numpy.zeros" title="View documentation for numpy.zeros">np.zeros</a>(n_drops, dtype=[('position', float, 2),

('size', float, 1),

('growth', float, 1),

('color', float, 4)])

# Initialize the raindrops in random positions and with

# random growth rates.

rain_drops['position'] = <a href="https://docs.scipy.org/doc/numpy/reference/generated/numpy.random.uniform.html#numpy.random.uniform" title="View documentation for numpy.random.uniform">np.random.uniform</a>(0, 1, (n_drops, 2))

rain_drops['growth'] = <a href="https://docs.scipy.org/doc/numpy/reference/generated/numpy.random.uniform.html#numpy.random.uniform" title="View documentation for numpy.random.uniform">np.random.uniform</a>(50, 200, n_drops)

# Construct the scatter which we will update during animation

# as the raindrops develop.

scat = ax.scatter(rain_drops['position'][:, 0], rain_drops['position'][:, 1],

s=rain_drops['size'], lw=0.5, edgecolors=rain_drops['color'],

facecolors='none')

def update(frame_number):

# Get an index which we can use to re-spawn the oldest raindrop.

current_index = frame_number % n_drops

# Make all colors more transparent as time progresses.

rain_drops['color'][:, 3] -= 1.0/len(rain_drops)

rain_drops['color'][:, 3] = <a href="https://docs.scipy.org/doc/numpy/reference/generated/numpy.clip.html#numpy.clip" title="View documentation for numpy.clip">np.clip</a>(rain_drops['color'][:, 3], 0, 1)

# Make all circles bigger.

rain_drops['size'] += rain_drops['growth']

# Pick a new position for oldest rain drop, resetting its size,

# color and growth factor.

rain_drops['position'][current_index] = <a href="https://docs.scipy.org/doc/numpy/reference/generated/numpy.random.uniform.html#numpy.random.uniform" title="View documentation for numpy.random.uniform">np.random.uniform</a>(0, 1, 2)

rain_drops['size'][current_index] = 5

rain_drops['color'][current_index] = (0, 0, 0, 1)

rain_drops['growth'][current_index] = <a href="https://docs.scipy.org/doc/numpy/reference/generated/numpy.random.uniform.html#numpy.random.uniform" title="View documentation for numpy.random.uniform">np.random.uniform</a>(50, 200)

# Update the scatter collection, with the new colors, sizes and positions.

scat.set_edgecolors(rain_drops['color'])

scat.set_sizes(rain_drops['size'])

scat.set_offsets(rain_drops['position'])

# Construct the animation, using the update function as the animation director.

animation = <a href="../../api/_as_gen/matplotlib.animation.FuncAnimation.html#matplotlib.animation.FuncAnimation" title="View documentation for matplotlib.animation.FuncAnimation">FuncAnimation</a>(fig, update, interval=10)

<a href="../../api/_as_gen/matplotlib.pyplot.show.html#matplotlib.pyplot.show" title="View documentation for matplotlib.pyplot.show">plt.show</a>()

</pre></div>

</div>

<a class="reference download internal" href="../../_downloads/rain.py" download=""><code class="xref download docutils literal notranslate">Download Python source code: rain.py</code></a></div>

<a class="reference download internal" href="../../_downloads/rain.ipynb" download=""><code class="xref download docutils literal notranslate">Download Jupyter notebook: rain.ipynb</code></a></div>

</div>

<a class="reference external" href="https://sphinx-gallery.readthedocs.io">Gallery generated by Sphinx-Gallery</a>

</div>

</div>

Last updated on Mar 06, 2018.

Created using <a

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

</div>

</body>

</footer>

</html>

matplotlib.github.com/2.2.0/gallery/animation/rain.html at main · matplotlib/matplotlib.github.com · GitHub

Expand file tree

Breadcrumbs

rain.html

Latest commit

History

Breadcrumbs

rain.html

File metadata and controls