Matplotlib如何实现关联输入变量的动态图例条目设置

本质是参数解析逻辑和plt.text不兼容 #

很多新手刚用legend的时候都会踩这个坑：

• plt.text() 接收的第三个位置参数就是要渲染的完整文本，你传格式化字符串进去会被当成一整段内容显示
• 但plt.legend()接收的第一个位置参数是标签列表：如果你直接丢一个字符串进去，它会把这个字符串当成可迭代的序列，逐个拆成单个字符，按顺序匹配给每一条已绘制的曲线当标签。
  举个实际运行的例子：你写plt.legend("Plot with input = 4")，因为你一共画了2条曲线，legend就会自动取字符串的前两个字符"P"、"l"当两条线的图例文字，和你想输入的动态内容完全不沾边，看着就像格式化逻辑没生效。

两种可靠的动态图例实现方式 #

1. 绘图时直接绑定动态label（推荐） #

这是最不容易出错的写法，直接在plt.plot阶段就把格式化好的动态文本绑定到对应曲线上，后续调用plt.legend()的时候会自动读取所有绑定的标签，不需要额外传参数：

from matplotlib import pyplot as plt
import numpy as np

input_var1 = 4
input_var2 = 3

y1 = np.random.rand(10)
y2 = np.random.rand(10)
x = np.linspace(0, 9, 10)

# 直接在plot里传动态格式化的label
plt.plot(x, y1, label=f"Plot with input = {input_var1}")
plt.plot(x, y2, label=f"Plot with input = {input_var2}")

# 空参调用legend即可自动匹配所有曲线的label
plt.legend()
plt.text(2, 0.2, "Some text with variable = {}".format(input_var1))   
plt.show()

2. 给legend传列表结构的标签 #

如果习惯先绘图、最后再统一设置图例，一定要注意传给legend的标签必须是列表/元组格式，顺序和你之前绘图的顺序一一对应：

plt.plot(x, y1)
plt.plot(x, y2)

# 把每个动态格式化好的标签按绘图顺序放进列表
plt.legend([
    f"Plot with input = {input_var1}",
    f"Plot with input = {input_var2}"
])

关键避坑提醒：哪怕你只画了1条线、只需要1个图例标签，也必须把格式化后的字符串包成列表再传给legend——直接传字符串一定会触发逐字符拆分的逻辑，根本没法正常显示完整的动态文本。

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