为Matplotlib多色线条图添加图例的问题

其实这不是Matplotlib的局限性，而是你用常规方法添加图例时没适配渐变线条的特性导致的。

为什么默认图例颜色不对？ #

常规的plt.plot()生成的是单一颜色的Line2D对象，Matplotlib的图例系统默认是基于这类单一样式的图形对象创建条目。但你用colormap绘制的沿长度变色的线条，一般是通过LineCollection（把线条拆成多段不同颜色的小线条）或者循环分段绘制实现的——这类线条本质是多个小线条的集合，不是单个统一样式的Line2D，所以默认图例只会取一个默认颜色（比如第一条小线条的颜色或者Matplotlib默认色），自然和整体渐变无关。

正确的解决方法 #

这里给你两种实用的方案：

方案1：手动创建自定义图例条目

如果你一定要用图例（而非色条），可以手动创建代表渐变关键区间的Line2D对象，把它们加到图例里：

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.collections import LineCollection
from matplotlib.lines import Line2D

# 生成示例数据
x = np.linspace(0, 10, 100)
y = np.sin(x)
points = np.column_stack((x, y))
segments = np.array([points[i:i+2] for i in range(len(points)-1)])

# 用colormap给线条上色
norm = plt.Normalize(x.min(), x.max())
lc = LineCollection(segments, cmap='viridis', norm=norm)
lc.set_array(x)
lc.set_linewidth(2)

fig, ax = plt.subplots()
ax.add_collection(lc)
ax.set_xlim(x.min(), x.max())
ax.set_ylim(y.min(), y.max())

# 创建自定义图例条目（精准匹配colormap颜色）
cmap = plt.get_cmap('viridis')
legend_elements = [
    Line2D([0], [0], color=cmap(norm(x.min())), lw=2, label='起始端'),
    Line2D([0], [0], color=cmap(norm(x.mean())), lw=2, label='中间段'),
    Line2D([0], [0], color=cmap(norm(x.max())), lw=2, label='结束端')
]

ax.legend(handles=legend_elements, loc='upper right')
plt.show()

方案2：用色条（Colorbar）替代图例

对于渐变线条，用色条来展示颜色和长度/数值的对应关系其实更直观，这也是Matplotlib推荐的方式：

# 接上面的LineCollection代码
fig.colorbar(lc, ax=ax, label='长度对应值')
plt.show()

色条会直接关联你的colormap和线条的颜色映射，比图例更清晰传达渐变的含义。

总结一下：渐变线条的图例问题是因为默认图例适配的是单一颜色线条，只要换用自定义图例条目或者色条，就能完美解决啦！

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