更新已绘制图形的颜色？

Question

我想知道是否可以更新一个参数，例如已经绘制的图形的线条颜色，它不会在破坏图形并创建另一个图形时换行。

示例：我绘制了一个图表，然后通过单击在其上创建了几条水平绿线。现在我想将图表的蓝色主线更改为红色，而不会丢失已创建的水平绿线。

类似：

import matplotlib.pyplot as plt

c = None
fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot([1,2,3],[1,2,3], color = c)

def onclick(event):
    plt.ion()
    plt.hlines(event.ydata,event.xdata-0.1,event.xdata+0.1,
           colors='green',linestyle='solid')  

cid = fig.canvas.mpl_connect('button_press_event', onclick)

def change_color():
    c = 'r'
    # ???

plt.show()

change_color() # running this function will update the plot line color to red

Answer 1

您需要捕获hlines 调用创建的艺术家：

fig, ax = plt.subplots()

arts = ax.hlines([.5, .75], 0, 1, lw=5)

返回一个LineCollection 对象。您可以通过编程方式对其进行修改

arts.set_color(['sage', 'purple'])

要更新窗口，您需要调用

fig.canvas.draw()

（在导入 pyplot 的 repl 上，master 上的最后一点不再正确）

我在这里做了一些更花哨的事情，并使用hlines 绘制了不止一条线并设置了不止一种颜色，但它也适用于只有一条线。