FilesExpand file tree

 master

/

basic_example_writer.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

43 lines (34 loc) · 1.15 KB

 master

/

basic_example_writer.py

Copy path

Top

File metadata and controls

• Code
• Blame

43 lines (34 loc) · 1.15 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

# Same as basic_example, but writes files using a single MovieWriter instance

# without putting on screen

# -*- noplot -*-

import numpy as np

import matplotlib

matplotlib.use("Agg")

import matplotlib.pyplot as plt

import matplotlib.animation as animation

def update_line(num, data, line):

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

return line,

# Set up formatting for the movie files

Writer = animation.writers['ffmpeg']

writer = Writer(fps=15, metadata=dict(artist='Me'), bitrate=1800)

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)

line_ani.save('lines.mp4', writer=writer)

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)

im_ani.save('im.mp4', writer=writer)