[Animated GIFs from Matplotlib Plots] #visualization

matplotlib_animation.py

import io

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


def create_plot(i):
    plt.clf()
    x = np.linspace(0, 10, 1000)
    y = np.sin((i + 1) * x)
    plt.plot(x, y)
    plt.title(f"Sine Wave (Frequency: {i+1})")
    plt.xlim(0, 10)
    plt.ylim(-1.5, 1.5)


# Create the animation
fig = plt.figure(figsize=(10, 6))

# Save frames as images
frames = []
for i in range(10):
    create_plot(i)
    buf = io.BytesIO()
    plt.savefig(buf, format="png")
    buf.seek(0)
    frames.append(Image.open(buf))

# Create and save the animated GIF
frames[0].save(
    "sine_wave_animation.gif",
    save_all=True,
    append_images=frames[1:],
    duration=200,
    loop=0,
)

print("Animation saved as 'sine_wave_animation.gif'")