python 绘制动图gif

import numpy as np
import gif
from matplotlib import pyplot as plt

if __name__ == '__main__':
    X = np.arange(0, 3+0.1, 0.1)
    imgs = []


    @gif.frame
    def plot_func(x, y_left, y_right, n):
        plt.figure(figsize=(5, 3), dpi=100)
        plt.plot(x, y_left, 60, c="r", alpha=0.7)
        plt.plot(x, y_right, 60, c="b", alpha=0.7)
        plt.title(f'n={n}')


    for n in range(1, 10):
        Y_left = (1 + X) ** n
        Y_right = 1 + X * n
        tmp = plot_func(X, Y_left, Y_right, n)
        imgs.append(tmp)
    gif.save(imgs, 'output.gif', duration=10, unit="s", between="startend")