Artist动画中动态修改标题与色条失效问题排查

核心问题 #

你只把图像对象（image）加入到了动画帧的艺术家列表中，但标题、色条这些动态变化的元素没有被包含进去。ArtistAnimation只会更新你传入列表里的艺术家对象，未被加入的元素会停留在最后一次设置的状态。

具体修正点 #

1. 捕获标题对象并加入帧列表
   不要用plt.title()，改用self.ax.set_title()获取返回的标题文本对象，将其加入当前帧的艺术家集合。
2. 将色条加入帧列表
   每次新建的色条对象也要加入当前帧的艺术家集合，确保动画能更新它。
3. 统一管理所有动态元素
   每一帧的所有需要变化的元素（图像、标题、色条）都要作为列表项加入self.images。

修改后的完整代码 #

from __future__ import division
 
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as anim

class AnimatedGif:
    def __init__(self, vmin=0, vmax=1):
        self.fig, self.ax = plt.subplots()
        self.images = []
        self.vmin = vmin
        self.vmax = vmax
        self.cb = None
        self.title = 'default title'
    
    def add(self, arr, vmin=None, vmax=None, title=None):
        if vmin is None:
            vmin = self.vmin
        if vmax is None:
            vmax = self.vmax
        if title is None:
            title = self.title
        
        # 创建图像对象
        image = plt.imshow(arr.T, vmin=vmin, vmax=vmax, animated=True)
        # 获取标题对象
        title_obj = self.ax.set_title(title)
        
        # 移除旧色条并创建新色条
        if self.cb is not None:
            self.cb.remove()
        self.cb = self.fig.colorbar(image)
        
        # 将所有动态元素加入当前帧的列表
        self.images.append([image, title_obj, self.cb])
    
    def save(self, filename):
        animation = anim.ArtistAnimation(self.fig, self.images)
        animation.save(filename, fps=1)

if __name__ == '__main__':
    arr = np.random.random((3,128,128))
    animated_gif = AnimatedGif()
    animated_gif.add(arr[0], vmin=0, vmax=0.3, title='image 0')
    for i in range(1, arr.shape[0]):
        animated_gif.add(arr[i], vmin=i*0.01, vmax=(i+1)*0.3, title='image {}'.format(i))
     
    animated_gif.save('arr.gif')

额外提示 #

如果动画播放时布局有偏移，可以在save方法前加上self.fig.tight_layout()，确保每帧的元素位置稳定。

内容的提问来源于stack exchange，提问作者Tamme