python - 在matplotlib中删除图形的特定部分

Question

我正在尝试执行以下操作:绘制两个不同的数据集，询问用户哪一个更好，从图中删除不好的数据集，然后保存好的图。但是，我找不到使用 matplotlib 来完成此操作的方法。我找到了 cla() 命令，但即使这样它仍然会删除整个图形...有人可以帮助我吗？

这是我的做法。

fig=plt.figure()
ax=fig.add_subplot(111)
plot(something)
ax2=fig.add_subplot(111)
plot(other thing)
cla()

谢谢!

Answer 1

这绘制了两条曲线。单击曲线会将其删除。

import matplotlib.pyplot as plt
import numpy as np
sin = np.sin
cos = np.cos
pi = np.pi

def delete(event):
    artist = event.artist
    artist.remove()
    event.canvas.draw()

fig = plt.figure()
ax = fig.add_subplot(111)

x = np.linspace(0, 2*pi, 50)
y = 2*sin(x)
z = 2*cos(x)
line1, = ax.plot(x,y)
line2, = ax.plot(x,z)
for artist in [line1, line2]:
    artist.set_picker(5)
fig.canvas.mpl_connect('pick_event', delete)

plt.show()