如何使用Python移除Matplotlib中的绘图？

嘿，我来帮你搞定Matplotlib移除绘图的问题，结合你给出的按键交互代码，这里有几种实用的方法，适配不同的场景：

1. 快速清除当前坐标轴的所有绘图 #

如果你的所有绘图都在同一个坐标轴上，想要一键清空所有线条、点等元素但保留坐标轴框架，可以用cla()方法：

import matplotlib.pyplot as plt

# 在你的按键事件处理里添加
plt.cla()  # 清除当前轴的所有绘图元素
plt.draw()  # 更新画布显示

要是你用的是面向对象的方式（比如创建了ax = plt.gca()），直接调用ax.cla()效果一样，这种方法适合快速重置当前轴的内容。

2. 清空整个画布 #

如果你的界面有多个子图，或者想完全重置整个绘图区域，就用clf()方法：

plt.clf()  # 清除整个画布的所有轴和绘图元素
plt.draw()

这个操作会把整个画布恢复到初始状态，适合需要彻底刷新的场景。

3. 精准移除特定绘图元素（适配你的按键交互场景） #

从你的代码来看，你是按'A'键逐步添加绘图的，大概率需要精准删除某一组添加的线条，而不是全部清空。这种情况最好的方式是保存绘图元素的引用，之后调用remove()方法删除。

具体改造你的代码： #

首先，在全局或类的作用域里维护一个列表，用来保存每次添加的绘图对象：

drawn_lines = []  # 保存所有添加的线条组
kern_list = []
recent_input_list = []
delete_list = []

然后在按'A'键添加绘图的部分，绘制时保存线条对象：

if event.key == 'A':
    xMax = 1500
    length = len(recent_input_list)
    if length == 0:
        x = [[0, 752, 1500], [248, 1251]]
        y = [[0, 1501, 0], [496, 495]]
        # 处理kern_list初始为空的情况
        kern = xMax + 300 if not kern_list else kern_list[0] + xMax + 300
        kern_list.insert(0, kern)
    else:
        kern = kern_list[0]
        print(kern)
        x = [[kern+0, kern+752, kern+1500], [kern+248, kern+1251]]
        y = [[0, 1501, 0], [496, 495]]
        kern = kern_list[0] + xMax + 300
        kern_list.insert(0, kern)
    recent_input_list.insert(length, event.key)
    delete_list.insert(length, event.key)
    
    # 绘制线条并保存对象
    current_group = []
    for i in range(len(x)):
        line, = plt.plot(x[i], y[i])
        current_group.append(line)
    drawn_lines.append(current_group)  # 把当前组线条加入列表
    plt.draw()  # 更新画布显示

最后，添加删除按键的处理（比如按'D'键删除最后添加的一组绘图）：

elif event.key == 'D':
    if drawn_lines:
        # 取出最后一组添加的线条并逐个移除
        last_group = drawn_lines.pop()
        for line in last_group:
            line.remove()
        # 同步维护你的其他列表
        if recent_input_list:
            recent_input_list.pop()
        if delete_list:
            delete_list.pop()
        if kern_list:
            kern_list.pop(0)  # 对应你添加时的insert(0)，这里pop(0)来同步
        plt.draw()  # 刷新显示

方法对比 #

• 用cla()/clf()适合快速清空，但无法精准控制删除哪一部分
• 保存绘图对象再remove()的方式，完全适配你的交互式添加/删除需求，能精准管理每一组添加的绘图元素

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