Matplotlib：如何清除图像流上的所有图形块（Patches）？

看起来你已经找对方向啦！要实现每次显示新图像前移除旧矩形的需求，核心就是在每次迭代开始时清除上一次添加到Axes中的所有矩形补丁。下面给你完善后的实现方案，以及关键细节的解释：

完整实现代码 #

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.patches import Rectangle

# 模拟你生成随机矩形的函数，替换成你自己的getListOfRandomRects即可
def getListOfRandomRects(n='random', dim=(800,800)):
    if n == 'random':
        n = np.random.randint(2, 10)
    rect_list = []
    img_h, img_w = dim
    for _ in range(n):
        # 生成随机的矩形坐标和尺寸，确保不超出图像范围
        x = np.random.randint(0, img_w - 50)
        y = np.random.randint(0, img_h - 50)
        width = np.random.randint(20, 50)
        height = np.random.randint(20, 50)
        rect_list.append((x, y, width, height))
    return rect_list

# 初始化画布和图像对象
fig, ax = plt.subplots(1)
im = ax.imshow(np.zeros((800, 800, 3)))
plt.ion()  # 开启交互模式，让图像能实时更新

# 维护一个列表，专门存储上一次绘制的矩形补丁，方便后续清除
previous_patches = []

for i in range(100):
    # 第一步：清除上一次的所有矩形
    for patch in previous_patches:
        patch.remove()  # 从Axes中移除补丁
    previous_patches.clear()  # 清空列表，准备存储新的补丁
    
    # 第二步：加载并更新新图像
    # 替换成你实际的图像读取代码
    img = plt.imread(f'my_image_{i:03d}.png')
    im.set_data(img)
    
    # 第三步：生成新矩形并添加到画布
    rect_lst = getListOfRandomRects(dim=img.shape[0:2])
    for rect in rect_lst:
        x, y, w, h = rect
        # 创建矩形补丁，这里可以根据你的需求调整样式
        rect_patch = Rectangle((x, y), w, h, edgecolor='red', facecolor='none', linewidth=2)
        ax.add_patch(rect_patch)
        previous_patches.append(rect_patch)  # 把新补丁加入列表
    
    # 第四步：更新画布，让新图像和矩形显示出来
    fig.canvas.draw()
    fig.canvas.flush_events()
    plt.pause(0.1)  # 可选：暂停一小段时间，方便观察效果

# 关闭交互模式，显示最终图像（如果需要）
plt.ioff()
plt.show()

关键细节解释 #

• 维护补丁列表：自己维护previous_patches列表是最安全的方式，能精准清除你绘制的矩形，不会误删Axes中其他可能存在的图形元素（比如后续添加的标注、线条）。
• 交互模式：plt.ion()开启交互模式后，不需要每次循环都重新创建figure，画布可以实时刷新，这对图像流的动态显示至关重要。
• 画布更新：每次添加新元素后，必须调用fig.canvas.draw()和fig.canvas.flush_events()来强制画布刷新，否则新的图像和矩形不会及时显示出来。

替代方案（快速清除所有补丁） #

如果你确定Axes中只有你绘制的矩形补丁，也可以直接清除ax.patches中的所有元素，代码更简洁：

# 替换清除旧补丁的代码块
while ax.patches:
    ax.patches[-1].remove()

不过这种方式风险较高，如果后续你在Axes中添加了其他图形，会被一起清除，所以更推荐第一种维护列表的方式。

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