Saving animations (GIF, MP4) in Matplotlib - Mini Project: Build & Apply

Saving animations (GIF, MP4)

📖 Scenario: You are creating a simple animated plot to show how a sine wave changes over time. You want to save this animation as both a GIF and an MP4 file so you can share it easily.

🎯 Goal: Build a Python script using matplotlib to create a sine wave animation and save it as a GIF and MP4 file.

📋 What You'll Learn

Create a time series data for the sine wave.

Set up the animation function using matplotlib.

Save the animation as a GIF file.

Save the animation as an MP4 file.

💡 Why This Matters

🌍 Real World

Animations help visualize changes over time in data, useful in science, engineering, and education.

💼 Career

Knowing how to create and save animations is valuable for data scientists and analysts who present dynamic data insights.

1

Create the sine wave data

Import numpy as np and matplotlib.pyplot as plt. Then create a numpy array called x with 100 points from 0 to 2π using np.linspace(0, 2 * np.pi, 100).

Matplotlib

import numpy as np
import matplotlib.pyplot as plt
# Create the x values for the sine wave
# Your code here

Hint

Use np.linspace to create evenly spaced points between 0 and 2π.

2

Set up the animation configuration

Create a figure and axis using plt.subplots() and assign them to variables fig and ax. Then create a line object using ax.plot with initial empty data and assign it to line.

Matplotlib

import numpy as np
import matplotlib.pyplot as plt
x = np.linspace(0, 2 * np.pi, 100)

fig, ax = plt.subplots()
# Create an empty line object on ax
# Your code here

Hint

Use ax.plot([], []) to create an empty line that will be updated in the animation.

3

Create the animation function

Import FuncAnimation from matplotlib.animation. Define a function called update that takes a frame number i and updates the line data to show y = np.sin(x + i / 10). Then create an animation object called ani using FuncAnimation with fig, update, and frames=100.

Matplotlib

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

x = np.linspace(0, 2 * np.pi, 100)
fig, ax = plt.subplots()
line, = ax.plot([], [])

def update(i):
    # Update the line data for frame i
    # Your code here

ani = FuncAnimation(fig, update, frames=100)

Hint

The update function changes the y-values of the line for each frame.

4

Save the animation as GIF and MP4

Save the animation ani as a GIF file named animation.gif using ani.save with writer='pillow'. Then save the same animation as an MP4 file named animation.mp4 using ani.save with writer='ffmpeg'. Finally, print 'Saved animation.gif and animation.mp4'.

Matplotlib

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

x = np.linspace(0, 2 * np.pi, 100)
fig, ax = plt.subplots()
line, = ax.plot([], [])

def update(i):
    y = np.sin(x + i / 10)
    line.set_data(x, y)
    return line,

ani = FuncAnimation(fig, update, frames=100)

# Save the animation as GIF and MP4
# Your code here

Hint

Use ani.save('filename', writer='pillow') for GIF and ani.save('filename', writer='ffmpeg') for MP4.

Practice

1. What is the correct method to save an animation created with matplotlib.animation.FuncAnimation as a file?

easy

A. Use anim.write(filename) to save the animation.

B. Use plt.savefig(filename) to save the animation.

C. Use animation.export(filename) to save the animation.

D. Use anim.save(filename) to save the animation.

Solution

1. Step 1: Understand animation saving method

   The FuncAnimation object has a method called save() specifically for saving animations.

2. Step 2: Differentiate from other save methods

   plt.savefig() saves static figures, not animations. There is no export() or write() method for animations in matplotlib.

3. Final Answer:

   Use anim.save(filename) to save the animation. -> Option D

4. Quick Check:

   Animation saving method = anim.save() [OK]

Hint: Remember: animation objects use save(), not plt.savefig() [OK]

Common Mistakes:

• Confusing plt.savefig() with anim.save()
• Trying to use non-existent methods like export() or write()
• Not calling save() on the animation object

2. Which writer should you specify in anim.save() to save an animation as a GIF file?

easy

A. 'pillow'

B. 'ffmpeg'

C. 'imagemagick'

D. 'avconv'

Solution

1. Step 1: Identify GIF writer options

   Matplotlib supports 'pillow' as the writer for saving GIF animations.

2. Step 2: Differentiate from other writers

   'ffmpeg' is used for MP4 videos, 'imagemagick' can also save GIFs but is less commonly used now, and 'avconv' is not a standard matplotlib writer.

3. Final Answer:

   'pillow' -> Option A

4. Quick Check:

   GIF writer = 'pillow' [OK]

Hint: Use 'pillow' writer for GIFs, 'ffmpeg' for MP4 videos [OK]

Common Mistakes:

• Using 'ffmpeg' for GIF saving
• Confusing 'imagemagick' as default GIF writer
• Not specifying any writer and expecting GIF output

3. What will happen if you run the following code snippet?

import matplotlib.pyplot as plt
import matplotlib.animation as animation

fig, ax = plt.subplots()
line, = ax.plot([], [])

def update(frame):
    line.set_data([0, frame], [0, frame**2])
    return line,

anim = animation.FuncAnimation(fig, update, frames=5)
anim.save('test_animation.mp4', writer='ffmpeg')

medium

A. The code will raise an error because 'ffmpeg' writer is not supported.

B. A GIF file named 'test_animation.mp4' will be created.

C. An MP4 video file named 'test_animation.mp4' will be created showing the animation.

D. Nothing will be saved because frames argument is invalid.

Solution

1. Step 1: Analyze animation creation and saving

   The code creates a simple animation with 5 frames and saves it as 'test_animation.mp4' using the 'ffmpeg' writer.

2. Step 2: Confirm writer and file type compatibility

   'ffmpeg' is the correct writer for MP4 files, so the file will be created successfully if FFmpeg is installed.

3. Final Answer:

   An MP4 video file named 'test_animation.mp4' will be created showing the animation. -> Option C

4. Quick Check:

   Saving MP4 with 'ffmpeg' = success [OK]

Hint: Use 'ffmpeg' writer to save MP4 animations [OK]

Common Mistakes:

• Expecting a GIF file with .mp4 extension
• Not having FFmpeg installed causing runtime error
• Misunderstanding frames argument as invalid

4. You try to save an animation as MP4 using anim.save('movie.mp4', writer='ffmpeg') but get an error: RuntimeError: ffmpeg not found. What is the best way to fix this?

medium

A. Change the writer to 'pillow' to save as MP4.

B. Install FFmpeg on your system and ensure it is in your PATH.

C. Rename the file to 'movie.gif' and save again.

D. Use plt.savefig() instead of anim.save().

Solution

1. Step 1: Understand the error cause

   The error means FFmpeg is not installed or not found in the system PATH, so matplotlib cannot use it to save MP4 files.

2. Step 2: Fix by installing FFmpeg

   Installing FFmpeg and adding it to the system PATH allows matplotlib to find and use it for saving MP4 animations.

3. Final Answer:

   Install FFmpeg on your system and ensure it is in your PATH. -> Option B

4. Quick Check:

   FFmpeg error fix = install FFmpeg [OK]

Hint: Install FFmpeg to fix 'ffmpeg not found' errors [OK]

Common Mistakes:

• Using 'pillow' writer for MP4 files
• Renaming file extension without changing writer
• Trying plt.savefig() which does not save animations

5. You want to save an animation as a GIF but also want to control the frame rate to 10 frames per second. Which of the following code snippets correctly saves the animation with these requirements?

import matplotlib.animation as animation

# anim is a FuncAnimation object
anim.save('animation.gif', ...)

hard

A. anim.save('animation.gif', writer='pillow', fps=10)

B. anim.save('animation.gif', writer='ffmpeg', fps=10)

C. anim.save('animation.gif', writer='pillow', frame_rate=10)

D. anim.save('animation.gif', fps=10)

Solution

1. Step 1: Identify correct writer for GIF

   Use 'pillow' as the writer to save GIF animations.

2. Step 2: Use correct parameter for frame rate

   The parameter to control frames per second is fps, not frame_rate.

3. Final Answer:

   anim.save('animation.gif', writer='pillow', fps=10) -> Option A

4. Quick Check:

   GIF save with fps uses writer='pillow' and fps=10 [OK]

Hint: Use writer='pillow' and fps=10 to save GIF at 10 fps [OK]

Common Mistakes:

• Using 'ffmpeg' writer for GIF files
• Using incorrect parameter name like frame_rate
• Omitting writer argument for GIF saving