如何使用Matplotlib将数组集合制作为带停顿的动画可视化？

要实现你的需求，咱们可以用Matplotlib自带的FuncAnimation模块来创建逐帧动画——它能帮你按时间序列切换不同数组对应的可视化效果，还能灵活设置帧之间的停顿时间。

先给你一个完整的修改后代码，咱们再一步步拆解说明：

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

plt.style.use('classic')

# 生成数组列表（和你的原逻辑一致，适配Python3语法）
a = []
for i in range(5):
    temp = np.zeros((5,5))
    temp[i, i] = 10
    a.append(temp)

# 预先加载图片为numpy数组，避免重复读取文件，提升动画流畅度
A_img = np.array(Image.open('A.png'))
B_img = np.array(Image.open('B.png'))

# 创建子图布局
fig, ax = plt.subplots(5, 5, figsize=(5, 5))
fig.subplots_adjust(hspace=0, wspace=0)

# 初始化所有子图的imshow对象，保存到列表方便后续更新
ims = []
for i in range(5):
    row_ims = []
    for j in range(5):
        # 隐藏坐标轴刻度，保持原可视化风格
        ax[i, j].xaxis.set_major_locator(plt.NullLocator())
        ax[i, j].yaxis.set_major_locator(plt.NullLocator())
        # 根据第一帧的数组值设置初始图像
        if a[0][i,j] == 10:
            im = ax[i, j].imshow(A_img, cmap="bone")
        else:
            im = ax[i, j].imshow(B_img, cmap="bone")
        row_ims.append(im)
    ims.append(row_ims)

# 定义动画更新函数：每帧切换对应数组的可视化
def update(frame):
    # 遍历所有子图，根据当前帧的数组值更新图像
    for i in range(5):
        for j in range(5):
            if a[frame][i,j] == 10:
                ims[i][j].set_data(A_img)
            else:
                ims[i][j].set_data(B_img)
    # 返回所有更新后的im对象，告诉动画哪些元素需要重绘
    return [im for row in ims for im in row]

# 创建动画：frames是总帧数，interval是帧之间的停顿时间（毫秒），repeat控制是否循环播放
anim = FuncAnimation(fig, update, frames=len(a), interval=1000, repeat=True, blit=True)

# 如果需要保存动画为GIF，取消下面的注释（需提前安装pillow库）
# anim.save('array_animation.gif', writer='pillow', dpi=100)

plt.show()

关键修改点说明： #

1. 引入动画核心模块：新增from matplotlib.animation import FuncAnimation，这是Matplotlib制作帧动画的标准工具。
2. 预加载图片优化：把A.png和B.png提前转换成numpy数组，避免每帧都重复读取文件，让动画播放更流畅。
3. 保存imshow对象：初始化时把每个子图的imshow实例存在二维列表里，后续更新时直接修改它们的set_data，不用重新创建子图，提升运行效率。
4. 帧更新逻辑：update(frame)函数接收当前帧的索引，根据对应数组a[frame]的值切换每个子图的图像，最后返回所有更新的元素（配合blit=True只重绘变化区域，进一步优化流畅度）。
5. 动画参数自定义：
   • frames=len(a)：动画总帧数和你的数组列表长度保持一致
   • interval=1000：每帧之间停顿1秒（可根据需求调整，单位为毫秒）
   • repeat=True：播放完所有帧后自动循环
   • blit=True：仅重绘变化的部分，减少不必要的计算

如果需要保存动画为GIF格式，确保已经安装pillow库（执行pip install pillow即可），或者换成writer='ffmpeg'（需提前安装ffmpeg工具）。

内容的提问来源于stack exchange，提问作者user491626