matplotlib GUIスレッドがフリーズするのを止めるには？

Question

from matplotlib import pyplot as p 
from scipy import zeros 
from Queue import Queue 
import random 

w,h = 320,200 

black = zeros((h,w,3), dtype='uint8') 
red = black.copy(); red[:,:,0] = 255 
green = black.copy(); green[:,:,1] = 255 
blue = black.copy(); blue[:,:,2] = 255 

def ants(): 
    from scipy import rand, dstack 
    return dstack([(255*rand(h,w)).astype('uint8')]*3) 

fig = p.figure() 
axs = [fig.add_subplot(1,3,i) for i in xrange(3)] 
[ax.imshow(black) for ax in axs] 

q = Queue() 

def update_image(ax): 
    ## this takes some time 
    import time 
    time.sleep(3) 
    ax.images[0].set_data(random.choice([red, green, blue])) 

def hit(event): 
    if event.inaxes in axs: 
    update_axs = [event.inaxes] 
    else: 
    update_axs = axs 
    for ax in update_axs: 
    ax.images[0].set_data(ants()) 
    p.draw() 
# for ax in update_axs: 
# update_image(ax) 
# p.draw() 

cid = fig.canvas.mpl_connect('button_press_event', hit) 
p.show() 

ここに私のコードは、すべて期待どおりに動作します。しかし、イベントハンドラで3行のコメントを外すと、私が期待していないいくつかの問題があります。最初にupdate_imageが動作している間にGUIがフリーズし、第2にdraw()への最初の呼び出しでは、update_imageが動作している間に、レースのアリを見ないのでペイントする機会を得られないようです。 matplotlibにこの種のものを設定するより良い方法は何ですか？それで、スムーズに動作しますか？

Answer 1

GUIがフリーズしないようにするには、別のスレッドまたはプロセスでupdate_imageを実行する必要があります。 threadingを使用して、あなたはこのような何かがあります

import matplotlib 
matplotlib.use('TkAgg') 
from matplotlib import pyplot as p 
from scipy import zeros 
import random 
import threading 


w,h = 320,200 

black = zeros((h,w,3), dtype='uint8') 
red = black.copy(); red[:,:,0] = 255 
green = black.copy(); green[:,:,1] = 255 
blue = black.copy(); blue[:,:,2] = 255 

def ants(): 
    from scipy import rand, dstack 
    return dstack([(255*rand(h,w)).astype('uint8')]*3) 

fig = p.figure() 
axs = [fig.add_subplot(1,3,i) for i in xrange(3)] 
[ax.imshow(black) for ax in axs] 

def update_image(ax): 
    ## this takes some time 
    import time 
    time.sleep(3) 
    ax.images[0].set_data(random.choice([red, green, blue])) 
    ax.figure.canvas.draw() 

def hit(event): 
    if event.inaxes in axs: 
     update_axs = [event.inaxes] 
    else: 
     update_axs = axs 
    for ax in update_axs: 
     ax.images[0].set_data(ants()) 
    p.draw() 
    for ax in update_axs: 
     t=threading.Thread(target=update_image,args=(ax,)) 
     t.daemon=True 
     t.start() 

cid = fig.canvas.mpl_connect('button_press_event', hit) 
p.show()