matplotlib如何删除一个点？

Question

问题：matplotlib如何删除一个点？

详细描述：
我用matplotlib在一张绘制有复杂曲线的画布上画了一个点，用scatter画的，但是我想添加一个删除这个点的功能。可是我发现，现有资料给出的clear()或cla()方法都是清空整个画布，然后再重新绘制曲线，以达到消除点的功能，但这样无意会加大程序运行的时间。

有没有只是删除某一个点的功能，仅仅是一个点，能找到删除scatter绘制的点的功能最好，我现在还是没有找到，不知各位有何高见？

已有的资料显示scatter这个方法绘制的点除了清空画布没有别的删除方法，真的是这样吗？如此成熟的一个绘图库，应该不会没有删除一个点的功能吧。

刚刚我找到了另一个绘制一个点的方法，即用ax_1.plot(x,y,marker='o',color='red')也能绘制一个点，不知这个方法是否有对应的删除一个点的功能呢？

Answer 1

引自免费微信小程序：皆我百晓生

要实现在matplotlib中仅移除特定的点而不清空整个画布，可以采用以下策略：首先，存储你绘制的点的句柄，然后在需要移除时调用其remove方法。这里，我们将使用scatter方法来演示这一过程。

下面是一段示例代码，展示了如何创建一个点，存储它的句柄，并在需要时移除它：

import matplotlib.pyplot as plt
import numpy as np

# 创建一个图和轴
fig, ax = plt.subplots()

# 生成一些数据
x_data = np.random.rand(10)
y_data = np.random.rand(10)

# 使用scatter绘制所有点，并保存句柄
points = ax.scatter(x_data, y_data)

# 选择要删除的点的索引
index_to_remove = 5

# 删除特定的点
if index_to_remove < len(points.get_offsets()):
    # 获取所有点的位置
    points.set_offsets(np.delete(points.get_offsets(), index_to_remove, axis=0))
    # 清除并重绘以更新图形
    fig.canvas.draw_idle()

plt.show()

上述代码的关键在于使用np.delete函数从points.get_offsets()返回的所有点坐标中移除指定索引的点。之后，通过调用points.set_offsets更新点的位置信息，并使用fig.canvas.draw_idle()强制图形更新。

对于使用plot方法绘制单个点的情况，同样可以存储返回的Line2D对象，并在需要时调用其remove方法。以下是如何操作的示例：

import matplotlib.pyplot as plt
import numpy as np

# 创建一个图和轴
fig, ax = plt.subplots()

# 绘制一个点并保存句柄
point, = ax.plot(1, 1, 'ro')

# 几秒后删除这个点
def remove_point():
    point.remove()
    fig.canvas.draw_idle()

# 假设我们想在几秒后删除这个点
import time
time.sleep(3)  # 等待3秒
remove_point()

plt.show()

在这个例子中，point是plot返回的Line2D对象列表中的第一个元素（因为我们只绘制了一个点）。通过调用point.remove()，我们可以从图表中移除这个点，再调用fig.canvas.draw_idle()来更新显示。

这两种方法都允许你在不重新绘制整个图表的情况下移除特定的点，从而提高程序效率。