python - 使用 matplotlib ArtistAnimation 将文本添加到动画图像

Question

我有几张图像作为二维数组，我想为这些图像创建动画并添加随图像变化的文本。

到目前为止，我已经成功制作了动画，但我需要你的帮助来向每张图片添加文本。

我有一个 for 循环来打开每个图像并将它们添加到动画中，假设我想将图像编号 (imgNum) 添加到每张图片。

这是我的代码，用于生成没有文本的图像电影。

ims = []
fig = plt.figure("Animation")
ax = fig.add_subplot(111)

for imgNum in range(numFiles):
    fileName= files[imgNum]

    img = read_image(fileName)

    frame =  ax.imshow(img)          

    ims.append([frame])

anim = animation.ArtistAnimation(fig, ims, interval=350, blit=True, repeat_delay=350)

anim.save('dynamic_images.mp4',fps = 2)

plt.show()

那么，如何向每张图片添加带有 imgNum 的文本？

感谢您的帮助!

Answer 1

您可以使用 annotate 添加文本并将 Annotation 艺术家添加到您传递给 ArtistAnimation 的艺术家列表中.这是一个基于您的代码的示例。

import matplotlib.pyplot as plt
from matplotlib import animation 
import numpy as np

ims = []
fig = plt.figure("Animation")
ax = fig.add_subplot(111)

for imgNum in range(10):
    img = np.random.rand(10,10) #random image for an example

    frame =  ax.imshow(img)   
    t = ax.annotate(imgNum,(1,1)) # add text

    ims.append([frame,t]) # add both the image and the text to the list of artists 

anim = animation.ArtistAnimation(fig, ims, interval=350, blit=True, repeat_delay=350)

plt.show()