Question

我有一个脚本绘制了一些测光孔径的数据，我想在xy图中绘制它们。我正在使用matplotlib.pyplot和python 2.5。

输入数据存储在大约500个文件中并读取。我知道这不是输入数据的最有效方式，但这是另一个问题......

示例代码：

import matplotlib.pyplot as plt

xcoords = []
ycoords = []

# lists are populated with data from first file

pltline, = plt.plot(xcoords, ycoords, 'rx')

# then loop populating the data from each file

for file in filelist:
    xcoords = [...]
    ycoords = [...]

pltline.set_xdata(xcoords)
pltline.set_ydata(ycoords)
plt.draw()

由于有超过500个文件，我偶尔会想要在绘图过程中关闭动画窗口。我绘制的代码可以正常工作，但它不会非常优雅地退出。单击关闭按钮时，绘图窗口不响应，我必须Ctrl+C。

任何人都可以帮助我找到一种在脚本运行时关闭动画窗口的方法，同时看起来优雅（比一系列python traceback错误更优雅）？

Answer 1

如果更新数据并在循环中进行绘制，则应该能够中断它。这是一个例子（绘制一个固定的圆圈，然后围绕周边移动一条线）：

from pylab import *
import time

data = []   # make the data
for i in range(1000):
    a = .01*pi*i+.0007
    m = -1./tan(a)
    x = arange(-3, 3, .1)
    y = m*x
    data.append((clip(x+cos(a), -3, 3),clip(y+sin(a), -3, 3)))


for x, y in data:  # make a dynamic plot from the data
    try:
        plotdata.set_data(x, y)
    except NameError:
        ion()
        fig = figure()
        plot(cos(arange(0, 2.21*pi, .2)), sin(arange(0, 2.21*pi, .2)))
        plotdata = plot(x, y)[0]
        xlim(-2, 2)
        ylim(-2, 2)
    draw()
    time.sleep(.01)

我输入time.sleep(.01)命令以确保我可以打破运行，但在我的测试中（运行Linux）没有必要。

优雅地退出matplotlib.pyplot动画

1 个答案: