循环中保存matplotlib pcolormesh图像时出现异常问题求助

你的代码出现两种异常的根源在于绘图元素的清理逻辑不对：

• 用plt.clf()时，它会清除整个Figure的所有内容，包括你在循环外创建的ax轴对象。第二次循环调用ax.pcolormesh()时，ax已经和当前Figure脱离，Matplotlib会自动新建轴，但布局会混乱，导致图像显示异常。
• 用ax.clear()时，它只清除轴内部的绘图内容（比如pcolormesh的网格），但之前生成的colorbar是属于Figure的独立元素，不会被清除，所以每次循环都会新增一个colorbar，最终出现多个colorbar叠加的问题。

下面是两种可行的解决办法：

方法一：每次循环重新创建绘图环境 #

把fig, ax = plt.subplots()放到循环内部，确保每次绘图都是全新的Figure和Axes，完全避免残留元素的干扰：

import matplotlib.pyplot as plt
import numpy as np

time_axis = np.arange(0, 30 + 1 / 50, 1 / 50)
another_axis = np.arange(0, 10, 1)

for i in range(2):
    grid = np.random.random((len(another_axis), len(time_axis)))
    # 每次循环新建fig和ax
    fig, ax = plt.subplots()
    pcm = ax.pcolormesh(time_axis, another_axis, grid)
    fig.colorbar(pcm, ax=ax)
    fig.savefig(f"test{i}.png")
    plt.close(fig)  # 关闭figure释放内存资源

方法二：复用绘图对象，手动管理colorbar #

如果想复用循环外创建的fig和ax，需要在每次循环时，除了清除轴内容，还要手动移除上一次的colorbar：

import matplotlib.pyplot as plt
import numpy as np

time_axis = np.arange(0, 30 + 1 / 50, 1 / 50)
another_axis = np.arange(0, 10, 1)

fig, ax = plt.subplots()
cb = None  # 用来记录上一次的colorbar对象

for i in range(2):
    grid = np.random.random((len(another_axis), len(time_axis)))
    ax.clear()  # 清除轴内的绘图元素
    pcm = ax.pcolormesh(time_axis, another_axis, grid)
    
    # 如果存在上一次的colorbar，先移除
    if cb is not None:
        cb.remove()
    # 创建新的colorbar并保存引用
    cb = fig.colorbar(pcm, ax=ax)
    
    fig.savefig(f"test{i}.png")
plt.close(fig)

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