Pyplot:刷新imshow()窗口

Question

我有两个方法: generate_window() (显示图像)和on_click() (对显示图像的窗口的单击作出反应)。它们看起来是这样的：

def generate_panel(img):
  plt.figure()
  ax = plt.gca()
  fig = plt.gcf()
  implot = ax.imshow(img)
  # When a colour is clicked on the image an event occurs
  cid = fig.canvas.mpl_connect('button_press_event', onclick)
  plt.show()

def onclick(event):
  if event.xdata != None and event.ydata != None:
    # Change the contents of the plt window here

在代码的最后一行，我希望能够更改plt窗口中显示的图像，但我似乎无法让它工作。我在不同的地方尝试过set_data()和that ()，但这没有效果。有什么建议吗？提前谢谢。

Answer 1

您还必须使用plt.ion()启用交互模式，然后在修改后只调用plt.draw()即可。

注意:在使用交互模式时，您必须在block=True上指定参数plt.show()，以防止它立即关闭窗口。

此示例的修改版本应在每次鼠标单击时绘制一个圆圈：

from matplotlib import pyplot as plt
import matplotlib.image as mpimg
import numpy as np


def generate_panel(img):
    plt.figure()
    ax = plt.gca()
    fig = plt.gcf()
    implot = ax.imshow(img)
    # When a colour is clicked on the image an event occurs
    cid = fig.canvas.mpl_connect('button_press_event', onclick)
    plt.show(block=True)


def onclick(event):
    if event.xdata is not None and event.ydata is not None:
        circle = plt.Circle((event.xdata,
                             event.ydata), 2, color='r')
        fig = plt.gcf()
        fig.gca().add_artist(circle)
        plt.draw()
        # Change the contents of the plt window here


if __name__ == "__main__":
    plt.ion()
    img = np.ones((600, 800, 3))
    generate_panel(img)