Créer une animation et la sauvegarder au format mp4 avec matplotlib

Exemple sur comment tracer une fonction cosinus et créer une animation avec matplotlib et python (source ). Pour connaitre le temps de la video il suffit de diviser le nombre d'images ici 100 par le nombre de frame par seconde ici 20 (fps=20), 100 / 20 = 5 secondes.

from random import randint

import numpy as np
import matplotlib
matplotlib.use("Agg")
import matplotlib.pyplot as plt
import matplotlib.animation as manimation

FFMpegWriter = manimation.writers['ffmpeg']
metadata = dict(title='Movie Test', artist='Matplotlib',
                comment='Movie support!')
writer = FFMpegWriter(fps=20, metadata=metadata)

fig = plt.figure()

x_min = 0
x_max = 20

with writer.saving(fig, "animation_cosinus.mp4", 100):
    for i in range(100):
        x = np.arange(x_min,i*x_max/100,0.1) 
        y = np.cos(x)
        plt.plot(x,y,color='red')
        plt.ylim(-1,1)
        plt.xlim(x_min,x_max)
        writer.grab_frame()

Recherches associées