使用watchdog动态更新绘图

Question

每当将新的txt文件添加到目录中时，我都希望绘制并显示该文件中的数据。如果出现另一个文件，我希望绘图更新并显示新数据。在main外部创建绘图会导致线程错误，因此我使用全局变量进行了修复(不是很好)。

问题是出现了一个白色的图形，而没有显示绘图。停止程序后，白色图形消失，绘图出现。正确的图像被保存到文件，但我希望图像是实时显示的。如果我注释掉plt.show()，则不会出现绘图。

我尝试了“在matplotlib中动态更新绘图”答案(Dynamically updating plot in matplotlib)，但发现因为它从未调用show()，所以没有出现任何窗口。如果我尝试调用show()，它会阻止更新。

插入plt.pause()不起作用(Real time matplotlib plot is not working while still in a loop)

示例代码不起作用(How to update a plot in matplotlib?)

下面是我的代码：

import time
from watchdog.observers import Observer
from watchdog.events import PatternMatchingEventHandler
import matplotlib.pyplot as plt
import numpy as np
import config

class MyHandler(PatternMatchingEventHandler):
    patterns=["*.txt", "*.TXT"]

    def on_created(self, event):
        self.process(event)

    def process(self, event):
        filename = event.src_path
        if '_AIt' in filename:
            config.isnew=True
            config.fname=filename


if __name__ == '__main__':
    observer = Observer()
    observer.schedule(MyHandler(), path='.', recursive=True)
    observer.start()

    dat=[0,1]
    fig = plt.figure()
    ax = fig.add_subplot(111)
    Ln, = ax.plot(dat)
    ax.set_xlim([0,10])
    ax.set_ylim([-1,1])
    plt.ion()
    plt.show()    

    try:
        while True:
            time.sleep(1)
            if config.isnew:
                config.isnew=False

                dataarray = np.array(np.transpose(np.loadtxt(config.fname)))
                dat = dataarray[15] #AI0
                Ln.set_ydata(dat)
                Ln.set_xdata(range(len(dat)))
                plt.savefig(config.fname[:-4] + '.png', bbox_inches='tight')
                plt.draw()

    except KeyboardInterrupt:
        observer.stop()
    observer.join()

config.py (创建配置设置的默认值)

isnew=False
fname=""

我之所以感到困惑，是因为下面的示例代码运行良好(来自pylab.ion() in python 2, matplotlib 1.1.1 and updating of the plot while the program runs)

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

dat=[0,1]
fig = plt.figure()
ax = fig.add_subplot(111)
Ln, = ax.plot(dat)
ax.set_xlim([0,20])
ax.set_ylim([0,40])
plt.ion()
plt.show()    

for i in range (18):
    dat=np.array(range(20))+i
    Ln.set_ydata(dat)
    Ln.set_xdata(range(len(dat)))
    plt.pause(1)

print 'done with loop'

Answer 1

以下代码应该可以执行您想要的操作：

import time
from watchdog.observers import Observer
from watchdog.events import PatternMatchingEventHandler
import matplotlib.pyplot as plt

plt.ion()  # enter interactive mode
ax = fig.add_subplot(111)
Ln, = ax.plot(dat)
ax.set_xlim([0,10])
ax.set_ylim([-1,1])
plt.draw()  # non-blocking drawing
plt.pause(.001)  # This line is essential, without it the plot won't be shown

try:
    while True:
        time.sleep(1)
        if config.isnew:
            ...
            plt.draw()
            plt.pause(.001)

except KeyboardInterrupt:
    observer.stop()
observer.join()

关键是在plt.draw调用之后调用plt.pause，否则它不会像其他python代码块matplotlib那样被绘制。值.001仅仅是一次尝试。我真的不知道它是如何在引擎盖下工作的，但这似乎是有效的。