在while循环中生成绘图后退出KeyboardInterrupt

时间:2013-01-07 16:31:06

标签: python windows matplotlib ipython enthought

我正在使用matplotlib实时监控实验,以在while循环中生成绘图。理想情况下,循环应该像KeyboardInterrupt之类的东西一样退出。这在Ubuntu测试中运行良好。在Windows 7中,使用ipython,它会以"Terminate batch job (Y/N)?"退出,然后关闭解释器。我想避免这种行为,并在KeyboardInterrupt之后保持解释器打开。这是一个测试脚本。

[编辑2]:如果ipython加载为ipython --pylab,此脚本在Windows中可以正常运行。

import time
import numpy as np
import matplotlib.pyplot as plt

fig = plt.figure()
ax = fig.add_subplot(111)
line, = ax.plot([0], [0], 'b-o')

window = 50
plot_data = np.zeros((window, 2))

i = 0
start = time.time()
while True:
    try:
        data = [time.time() - start, np.random.rand()]
        print ' '.join('{:.2f}'.format(x) for x in data)
        if i < window:
            plot_data[i,:] = data
            line.set_data(plot_data[0:i+1,0], plot_data[0:i+1,1])
        else:
            plot_data[0:window-1] = plot_data[1:window]
            plot_data[window-1] = data
            line.set_data(plot_data[:,0], plot_data[:,1])
        ax.relim()
        ax.autoscale_view(True,True,True)
        fig.canvas.draw()
        plt.pause(0.1)
        i += 1
    except KeyboardInterrupt:
            print "Program ended by user.\n"
            break 
print 'Success!'

[编辑1]:我应该更清楚为什么用matplotlib标记这个。以下示例脚本在任一操作系统中都不会出现问题。

i = 0 
start = time.time()
while True:
    try:
        data = [time.time() - start, np.random.rand()]
        print ' '.join('{:.2f}'.format(x) for x in data)
        time.sleep(0.1)
    except KeyboardInterrupt:
        print "Proram ended by user. \n"
        break
print 'Success!'

昨天安装了所有软件包,作为Enthought

的全新安装的一部分

2 个答案:

答案 0 :(得分:1)

现在,我发现在多台Windows机器上解决此问题的最佳方法如下......

print 'press \'q\' to end run'
time.sleep(1.0)

fig = plt.figure()
ax = fig.add_subplot(111)
line, = ax.plot([0], [0], 'b-o')

window = 150
plot_data = np.zeros((window, 2))

i = 0
start = time.time()
while True:
    data = [time.time() - start, np.random.rand()]
    print ' '.join('{:.2f}'.format(x) for x in data)
    if i < window:
        plot_data[i,:] = data
        line.set_data(plot_data[0:i+1,0], plot_data[0:i+1,1])
    else:
        plot_data[0:window-1] = plot_data[1:window]
        plot_data[window-1] = data
        line.set_data(plot_data[:,0], plot_data[:,1])
    ax.relim()
    ax.autoscale_view(True,True,True)
    fig.canvas.draw()
    plt.pause(delay)
    i += 1
    if msvcrt.kbhit():
        if ord(msvcrt.getch()) == 113:
            print "Program ended by user.\n"
            break 
print 'Success!'

不幸的是,这根本不是平台独立的,但过去几天我读过的所有内容都让我相信平台无关的键盘输入实际上是不可能实现的。我原来问题中的代码在Unix和一些Windows安装中运行良好。此代码适用于我尝试过的少数Windows安装。所有这些在ipython --pylab运行时效果最佳。这可能现在已经足够好了。

答案 1 :(得分:0)

我无法重现你的问题。我将测试脚本复制到文件ipythonexception.py中,并使用ipython命令从run启动它。然后我得到以下输出。

In [1]: run ipythonexception.py
0.00 0.27
0.31 0.11
0.56 0.44
0.81 0.26
1.06 0.27
1.29 0.66
1.53 0.32
1.76 0.74
2.03 0.44
2.26 0.89
Program ended by user.

Success!

In [2]: window
Out[2]: 50

In [3]:

一切似乎都应该发挥作用。我在Windows 7Python 2.7.2Ipython 0.13

您是否使用批处理文件启动Ipython?我叮叮"Terminate batch job (Y/N)?"是一条Windows消息,如果我打破一些批处理作业,我通常会收到这条消息。