2016-09-26 58 views
0

餘米試圖在我的互動情節 的中間增加一個註解,我想看到我的循環值產生我測試列表 與我的所有數據添加註釋。對於每個imshow情節,我想看到我的值, 我添加ax.annotate,但它不起作用。在交互式情節

import numpy as np 
import matplotlib.pyplot as plt 
import matplotlib.animation as animation 
fig = plt.figure() # make figure 
ax = fig.add_subplot(111) 

test = [] 
mask2 = np.random.randint(255, size=(20, 20)) 

for i in range(1,5,3): 
    kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE,(i,i)) 
    res = (cv2.morphologyEx(mask2.astype(uint8),cv2.MORPH_OPEN,kernel)) 
    #plt.imshow(res,cmap=plt.cm.gray,alpha=1);plt.show() 
    test.append(res) 


# make axesimage object 
# the vmin and vmax here are very important to get the color map correct 
im = ax.imshow(test[0], cmap=plt.get_cmap('hot'), vmin=0, vmax=255) 
im2 = ax.annotate('This is awesome!', 
      xy=(76, -10.75), 
      xycoords='data', 
      textcoords='offset points', 
      arrowprops=dict(arrowstyle="->")) 
plt.show() 

# function to update figure 
def updatefig(j): 
    # set the data in the axesimage object 
    im.set_array(test[j]) 
    # return the artists set 
    return im, 
# kick off the animation 
ani = animation.FuncAnimation(fig, updatefig, frames=range(20), 
           interval=50, blit=True) 
plt.show() 

回答

1

我找到了出路。我在我的更新功能裏添加了一個「set_text」,我返回圖片和文字:

test = [] 
    test2 = [] 
    for i in range(3,27,3): 
kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE,(i,i)) 
res = (cv2.morphologyEx(mask2,cv2.MORPH_OPEN,kernel)) 
#plt.imshow(res,cmap=plt.cm.gray,alpha=1);plt.show() 
test.append(res) 
test2.append(i) 


    fig = plt.figure() # make figure 
    ax = fig.add_subplot(111) 

    # make axesimage object 
    # the vmin and vmax here are very important to get the color map correct 
    im = ax.imshow(test[0], cmap=plt.get_cmap('hot'), vmin=0, vmax=255) 
    time_template = 'Diffusion - Kernel size : %2.2d' # prints running simulation time 
    txt = ax.text(500, 80, '', fontsize=15,color='red') 
    #plt.show() 

    # function to update figure 
    def updatefig(j): 
    # set the data in the axesimage object 
    im.set_array(test[j]) 
    txt.set_text(time_template%(float(np.asarray(test2[j])))) 
    return im,txt 

    ani = animation.FuncAnimation(fig, updatefig, frames=range(len(test)), 
       interval=100, blit=False,repeat=True) 
    plt.show()