如何动态更新Matplotlib图表的图例/标签？

场景1：每次重绘时更新图例标签 #

如果你的代码是在循环或函数调用中重复生成数据并绘图，直接在plot函数里用格式化字符串插入动态的alpha值即可。假设你有一个返回整数的函数（比如get_current_alpha()），每次绘图时把它的值嵌入label：

import matplotlib.pyplot as plt

# 示例：返回动态alpha值的函数，替换为你的实际逻辑
def get_current_alpha():
    # 这里可以是从外部参数、状态或计算中获取的动态值
    return 5

# 每次需要更新绘图时执行以下代码
plt.clf()  # 清除当前画布，避免图形重叠
vals = [1,2,3]  # 替换为你的实际数据
smpl_sz = [10,20,30]
val2 = [1,2,3]
smpl_sz2 = [15,25,35]

current_alpha = get_current_alpha()
plt.plot(vals, smpl_sz, 'b', marker='o', label=f'alpha={current_alpha}')
plt.plot(val2, smpl_sz2, 'g', marker='o', label=f'alpha={current_alpha}')

plt.title('Power vs Sample Size')
plt.xlabel('Power')
plt.ylabel('Sample Size')
plt.legend(loc="lower right")
plt.draw()  # 刷新显示（交互式环境下需要）

场景2：不重绘整个图，仅更新已有图例标签 #

如果已经绘制了线条，想要在不重新绘制整个图的情况下更新图例，可通过修改线条对象的label属性，再更新图例：

import matplotlib.pyplot as plt
plt.ion()  # 开启交互式模式，支持实时更新

# 初始化绘图并保存线条对象
vals = [1,2,3]
smpl_sz = [10,20,30]
val2 = [1,2,3]
smpl_sz2 = [15,25,35]

line1, = plt.plot(vals, smpl_sz, 'b', marker='o')
line2, = plt.plot(val2, smpl_sz2, 'g', marker='o')
plt.title('Power vs Sample Size')
plt.xlabel('Power')
plt.ylabel('Sample Size')
leg = plt.legend(loc="lower right")
plt.show(block=False)  # 非阻塞显示窗口，允许后续更新

# 模拟alpha值动态变化的过程
for new_alpha in [3, 5, 7, 9]:
    # 更新两条线的label
    line1.set_label(f'alpha={new_alpha}')
    line2.set_label(f'alpha={new_alpha}')
    # 更新图例内容
    leg.update_labels([line1, line2])
    # 若需要同步更新数据，可调用line1.set_data(new_vals, new_smpl_sz)
    plt.draw()
    plt.pause(1)  # 暂停1秒，便于观察变化

关键要点 #

• 使用f-string格式化（f'alpha={current_alpha}'）是最简洁的插入动态变量的方式。
• 交互式绘图需开启plt.ion()，并通过plt.draw()或plt.pause()刷新显示。
• 仅更新图例时，必须保存线条对象和图例对象，通过set_label()和update_labels()方法修改。

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