使用Seaborn基于预填充数组制作动态热力图失败，请求协助

问题根源 #

• 每次调用sns.heatmap都会创建新的轴对象，导致热力图不断叠加而非更新现有内容
• init和animate函数未返回需要动画更新的艺术家对象，Matplotlib无法追踪更新元素
• 数据值设置不合理：i/2在i≥2时就超过了vmax=.8，后续帧颜色不会产生变化

修复后的代码 #

import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt
from matplotlib import animation

%matplotlib notebook
plt.rcParams["figure.figsize"] = [5, 5]
plt.rcParams["figure.autolayout"] = True

# 调整数据范围，确保所有值在vmax限制内，保证颜色变化可见
mydata = np.zeros((20,5,5))
for i in range(20):
    mydata[i] = (i/20)*0.8  # 值从0到0.8线性递增

fig = plt.figure()
# 初始化固定轴对象，后续复用避免重复创建
ax = plt.gca()

def init():
    # 绘制初始热力图，绑定到固定轴
    heatmap = sns.heatmap(mydata[0], vmax=.8, cbar=False, square=True, ax=ax)
    return heatmap.get_children()

def animate(i):
    # 清除当前轴内容，防止热力图叠加
    ax.clear()
    # 在同一轴上重新绘制当前帧的热力图
    heatmap = sns.heatmap(mydata[i], vmax=.8, cbar=False, square=True, ax=ax)
    return heatmap.get_children()

anim = animation.FuncAnimation(fig, animate, init_func=init, frames=20, repeat=False)

关键修改点 #

• 复用轴对象：创建固定的ax对象，所有热力图绘制都绑定到该轴，避免生成冗余轴
• 清除旧内容：animate函数开头用ax.clear()清除上一帧的热力图，防止叠加显示
• 返回更新元素：init和animate返回热力图的子元素，让Matplotlib能追踪需要更新的部分
• 调整数据范围：修改数据生成逻辑，确保所有帧的值都在vmax范围内，保证颜色变化效果

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