如何阻止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行時，有一些問題我沒有想到。首先GUI凍結，而update_image正在工作，其次第一個電話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()