在for循环中重复使用seaborn画图，未清除缓存导致一张图中存在多条colorbar

最新推荐文章于 2025-08-01 15:52:43 发布

hon·ey·su·ckle

最新推荐文章于 2025-08-01 15:52:43 发布

阅读量574

点赞数 2

CC 4.0 BY-SA版权

分类专栏： python 文章标签： python

本文链接：https://blog.csdn.net/qq_41599380/article/details/131120227

python 专栏收录该内容

3 篇文章

订阅专栏

在尝试用for循环绘制21张2*20热力图时，遇到每张图叠加前一张的颜色柱问题。原因是未清除图像缓存。通过添加plt.close(all)到循环中，每次绘制后关闭所有图形窗口，成功解决了颜色柱积累的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题描述

有一个 $21 * 2 * 20$ 的矩阵，分别代表21个类别的数据，每个类别的输出是一个 $2 * 20$ 的矩阵，我想要把每个类别的输出用热力图画出来，一共需要画21张图。
因此使用了for循环，代码如下：

avg_out_open = np.random.rand(21, 2, 20)
for i, open_vector in enumerate(avg_out_open):
    fig = sns.heatmap(data=open_vector,cmap="RdBu_r",linewidths=0.3) 
    heatmap = fig.get_figure()
    heatmap.savefig(log_name+str(step)+'out_open_class'+str(i)+'.png', dpi = 400)

但是这样的代码存在问题。
画出的图中会有多条颜色柱，并且后面的循环颜色柱越多。
猜测是由于没有清除缓存导致的，尝试使用fig.cla()，无效。
在这里插入图片描述

解决办法

使用plt.close("all")

for i, open_vector in enumerate(avg_out_open):
    fig = sns.heatmap(data=open_vector,cmap="RdBu_r",linewidths=0.3) 
    heatmap = fig.get_figure()
    heatmap.savefig(log_name+str(step)+'out_open_class'+str(i)+'.png', dpi = 400)
    plt.close("all")