Python將文本添加到matplotlib.image.AxesImage實例

Question

我有這段代碼用於從一系列矩陣創建一系列圖像，並且在每個圖像中我想添加特定文本。這是我的典型代碼：

ax = axes([0,0,1,1]) 

for i in range(0,5): 
    text(1,1,str(i)) 
    ax.imshow(a[:,:,i],origin='lower') 
    savefig(str("%04d" % int(i))+'.png',format="png") 
    del ax.texts[-1] 

但問題是，隨着迭代數量的增加，速度死亡，它變得如此如此緩慢。似乎在後臺打開很多窗口是有問題的。

有什麼建議嗎？

Answer 1

我只是說這一條線上的循環結束，現在工作得很好。這只不過是在記憶中積累了前所未有的數字的問題。

ax = axes([0,0,1,1]) 

for i in range(0,5): 
    text(1,1,str(i)) 
    ax.imshow(a[:,:,i],origin='lower') 
    savefig(str("%04d" % int(i))+'.png',format="png") 
    del ax.texts[-1] 
    close(gcf()) 

Answer 2

而不是創建一個新的圖像和文本對象每個循環重用對象。

ax = axes([0,0,1,1]) 
t = text(1,1,str(0)) 
img = ax.imshow(a[:,:,0],origin='lower') 

for i in range(0,5): 
    t.set_text(str(i) 
    img.set_data(a[:,:,i]) 
    savefig(str("%04d" % int(i))+'.png',format="png") 

也看到

Visualization of 3D-numpy-array frame by frame