Python使用Seaborn动态更新热图数据时出现图层嵌套问题求解

每次循环调用sns.heatmap()时，函数默认会在当前活跃的绘图坐标轴上叠加新的热图图层，不会自动清空之前的绘制内容，因此会出现热图嵌套的问题。

核心修改点如下：

• 初始化阶段就创建全局唯一的画布与坐标轴对象，避免重复创建
• 每次绘制新热图前，先清空坐标轴上的已有内容
• 可选配置：固定热图的数值上下限，保证所有帧的色阶基准统一，避免颜色频繁跳动

修改后的可运行代码如下：

import numpy as np
np.random.seed(0)
import seaborn as sns
import matplotlib.pyplot as plt

# 提前创建全局唯一的画布和坐标轴，仅初始化一次
fig, ax = plt.subplots()
data = np.random.rand(120 ,5900)
data_to_draw = np.zeros(shape=(1,5900))
for i,d in enumerate(data):
    # 更新待绘制数据
    data_to_draw = np.vstack((data_to_draw, data[i]))
    # 最多保留5行显示
    if data_to_draw.shape[0]>5:
        data_to_draw = data_to_draw[1:]
    
    # 清空坐标轴上的旧内容
    ax.cla()
    # 绘制热图，固定色阶范围避免颜色跳动，指定在提前创建的ax上绘制
    sns.heatmap(data_to_draw, cmap="coolwarm", vmin=0, vmax=1, ax=ax)

    plt.draw()
    plt.pause(0.1)

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