matplotlib.github.com/examples/animation/basic_example.py at master · jmg6yo/matplotlib.github.com · GitHub

46 lines (35 loc) · 1.17 KB

"""

=========================

Simple animation examples

=========================

This example contains two animations. The first is a random walk plot. The

second is an image animation.

"""

import numpy as np

import matplotlib.pyplot as plt

import matplotlib.animation as animation

def update_line(num, data, line):

line.set_data(data[..., :num])

return line,

fig1 = plt.figure()

data = np.random.rand(2, 25)

l, = plt.plot([], [], 'r-')

plt.xlim(0, 1)

plt.ylim(0, 1)

plt.xlabel('x')

plt.title('test')

line_ani = animation.FuncAnimation(fig1, update_line, 25, fargs=(data, l),

interval=50, blit=True)

# To save the animation, use the command: line_ani.save('lines.mp4')

fig2 = plt.figure()

x = np.arange(-9, 10)

y = np.arange(-9, 10).reshape(-1, 1)

base = np.hypot(x, y)

ims = []

for add in np.arange(15):

ims.append((plt.pcolor(x, y, base + add, norm=plt.Normalize(0, 30)),))

im_ani = animation.ArtistAnimation(fig2, ims, interval=50, repeat_delay=3000,

blit=True)

# To save this second animation with some metadata, use the following command:

# im_ani.save('im.mp4', metadata={'artist':'Guido'})

plt.show()

Expand file tree