如何利用Matplotlib的ArtistAnimation动态更新图例？

原代码的核心问题是：

• 每次循环调用ax.legend()会创建新图例对象，所有图例叠加在同一位置，后创建的图例遮挡之前的；
• ArtistAnimation仅管理容器中艺术家的可见性，但叠加的图例导致只有最上层的（最后一个绿色）图例能被看到，前两帧的图例被遮挡无法显示。

解决方案1：修正ArtistAnimation代码 #

通过手动控制图例的绘制层级，确保当前帧的图例处于最上层：

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

a = np.random.rand(3,3)
fig, ax=plt.subplots()
container = []
colors = ["red","blue","green"]
labels = [0,1,2]

max_zorder = 10  # 初始层级基准

for i in range(a.shape[1]):
    line, = ax.plot(a[:,i], color=colors[i])
    line.set_visible(False)
    
    title = ax.text(0.5,1.05,"Title {}".format(i), 
                    size=plt.rcParams["axes.titlesize"],
                    ha="center", transform=ax.transAxes, )
    title.set_visible(False)
    
    # 创建独立图例并手动添加到轴，设置初始层级
    patch = mpl.patches.Patch(color=colors[i], label=labels[i])
    legend = mpl.legend.Legend(ax, [patch], [labels[i]], bbox_to_anchor=(1.05, 1), loc='upper left')
    legend.set_zorder(max_zorder + i)
    legend.set_visible(False)
    ax.add_artist(legend)
    
    container.append([line, title, legend])

# 自定义帧更新逻辑，确保当前帧图例层级最高
def update_frame(frame_idx):
    # 隐藏所有帧的元素
    for frame_artists in container:
        for artist in frame_artists:
            artist.set_visible(False)
    # 显示当前帧的元素
    current_artists = container[frame_idx]
    for artist in current_artists:
        artist.set_visible(True)
        # 提升当前图例层级到最高
        if isinstance(artist, mpl.legend.Legend):
            artist.set_zorder(max_zorder + len(colors))

# 替换ArtistAnimation默认的帧绘制逻辑
ani = animation.ArtistAnimation(fig, container, interval=750, blit=False)
ani._draw_frame = update_frame

plt.show()

解决方案2：改用FuncAnimation（更简洁高效） #

仅维护一组元素，每次帧更新时直接修改属性，从根源避免图例叠加问题：

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

a = np.random.rand(3,3)
fig, ax=plt.subplots()
colors = ["red","blue","green"]
labels = [0,1,2]

# 初始化基础元素
line, = ax.plot(a[:,0], color=colors[0])
title = ax.text(0.5,1.05,"Title {}".format(0), 
                size=plt.rcParams["axes.titlesize"],
                ha="center", transform=ax.transAxes)
legend_patch = mpl.patches.Patch(color=colors[0], label=labels[0])
legend = ax.legend(handles=[legend_patch])

def update(frame):
    # 更新线条数据与颜色
    line.set_data(range(a.shape[0]), a[:,frame])
    line.set_color(colors[frame])
    # 更新标题文本
    title.set_text(f"Title {frame}")
    # 更新图例（移除旧图例，创建新图例）
    global legend
    legend.remove()
    new_patch = mpl.patches.Patch(color=colors[frame], label=labels[frame])
    legend = ax.legend(handles=[new_patch])
    return line, title, legend

ani = animation.FuncAnimation(fig, update, frames=3, interval=750, blit=False)
plt.show()

说明 #

• 方案1适配ArtistAnimation的特性，通过层级调整解决叠加问题，但代码相对复杂；
• 方案2使用FuncAnimation直接更新元素属性，逻辑更清晰，是更推荐的实现方式。

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