PyCharm matplotlib 交互式图形不阻塞执行

Question

我阅读了很多关于此主题的堆栈溢出问题，但经过大量实验后，我无法弄清楚我的问题。

我在 Windows 7 上使用 PyCharm 2016.3.2（但我在 OSX 上遇到同样的问题），我的解释器是 Python 3.6 和 matplotlib 2.0.0 的最新版本的 Anaconda。

这是我试图实现的目标（也许我没有使用正确的方法，因为我试图重新创建我在 Octave/Matlab 中习惯的行为）：

1. 在弹出窗口中绘制一个图形
2. 暂停我的脚本（input("press a key to continue")）
3. 观察图形，然后按一个键继续脚本
4. 计算其他东西
5. 在同一个图上绘制新数据
6. 暂停我的脚本（input("press a key to continue")）
7. 观察图形，然后按一个键继续脚本

这是我的测试代码：

import matplotlib
matplotlib.use('TkAgg')
import matplotlib.pyplot as plt

print('Plotting Data...')

plt.ion()
plt.figure(1)
plt.plot([1, 5, 10, 20], [1, 5, 10, 20])
plt.xlabel('x label')
plt.ylabel('y label')
plt.show()
plt.pause(0.0001)

input('Plotting Data done..., Press a key to continue')

plt.figure(1)
plt.plot([1, 5, 10, 20], [2, 10, 20, 40])
plt.show()
plt.pause(0.0001)

input('Program paused. Press enter to end.\n')

这是我想要的最接近的版本，当我将鼠标悬停在图上时，图是正确的但没有响应（plt.pause(0.0001) 会生成警告，但代码有效）。

我经常使用参数 (plt.ion() ; plt.pause() ; plt.show(block=False))。 大多数情况下，这会导致绘图窗口为空，或者我需要关闭窗口才能继续执行。

感谢您的帮助！

Answer 1

一旦plt.show() 被调用，主循环就会被窗口接管。因此，一旦调用了input，主循环就会变得无响应。

您可以尝试停留在 GUI 循环中并在那里处理按键操作。

import matplotlib
matplotlib.use('TkAgg')
import matplotlib.pyplot as plt

print('Plotting Data...')

fig, ax = plt.subplots()
ax.set_xlabel('x label')
ax.set_ylabel('y label')
i=[0]

def f1():
    ax.plot([1, 5, 10, 20], [1, 5, 10, 20])

def f2():
    ax.plot([1, 5, 10, 20], [2, 10, 20, 40])

def f3():
    ax.plot([1, 5, 10, 20], [5, 9, 17, 28])


def update(event=None):
    if i[0]==0: f1()
    if i[0]==1: f2()
    if i[0]==2: f3()
    fig.canvas.draw_idle()
    i[0]+=1
    print('Step {} done..., Press a key to continue'.format(i[0]))

fig.canvas.mpl_connect("key_press_event", update)    

update()
plt.show()