如何在python中制作包含多个不同动画的gif或短mp4视频？

from PIL import Image
import imageio
import numpy as np
import matplotlib.pyplot as plt

# 创建动画帧
frames = []
for i in range(10):
    # 创建一个新的图像
    img = Image.new('RGB', (200, 200), color=(255, 255, 255))
    draw = ImageDraw.Draw(img)
    
    # 在图像上绘制动画内容
    draw.rectangle([(50, 50), (150, 150)], fill=(255, 0, 0))
    draw.text((80, 80), f'Frame {i}', fill=(0, 0, 0))
    
    # 将图像转换为numpy数组
    frame = np.array(img)
    frames.append(frame)

# 将帧保存为图像文件
for i, frame in enumerate(frames):
    imageio.imwrite(f'frame_{i}.png', frame)

# 将图像文件转换为视频
images = []
for i in range(10):
    images.append(imageio.imread(f'frame_{i}.png'))
imageio.mimsave('animation.gif', images, fps=10)

# 保存为短MP4视频
imageio.mimsave('animation.mp4', images, fps=10, plugin='ffmpeg')