如何绘制并导出边缘无断点的平滑多色线条

嘿，我之前也碰到过这个坑——用LineCollection画多色线在plt.show()里看着好好的，导出PDF就出现首尾断点，太闹心了。结合你的需求（无断点平滑线条，接受颜色平滑过渡），给你几个实测有效的解决办法：

方法1：给LineCollection加对样式参数（最推荐） #

其实核心问题是LineCollection的每条线段默认是独立渲染的，PDF后端对线段连接处的处理和plt.show()用的Agg后端不一样。只要设置对两个关键参数，就能搞定断点：

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

# 生成示例数据：正弦曲线拆分成线段
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)])

# 生成从蓝到红的渐变颜色
colors = plt.cm.viridis(np.linspace(0, 1, len(segments)))

# 重点：设置joinstyle和capstyle，同时保证线宽足够
lc = LineCollection(
    segments,
    colors=colors,
    linewidth=4,
    joinstyle='round',  # 相邻线段连接处圆润，消除间隙！
    capstyle='round',  # 整条线的首尾端点圆润
    antialiased=True,
    linestyle='solid'
)

fig, ax = plt.subplots()
ax.add_collection(lc)
ax.autoscale()

# 导出PDF时用高DPI，避免模糊
plt.savefig('smooth_multicolor_line.pdf', dpi=300, bbox_inches='tight')
plt.show()

划重点： #

• joinstyle='round'是解决中间断点的关键，它会让线段连接处自动过渡，不会出现缝隙；
• capstyle='round'只影响整条线的首尾端点，对中间线段的连接处没用；
• 如果你的matplotlib版本太老（比如低于3.3.0），PDF后端可能不支持这些样式，建议升级到最新版。

方法2：让线段轻微重叠（极端情况备用） #

要是方法1不管用，可能是你的PDF渲染器对样式支持有问题，这时候可以让相邻线段重叠一点，从物理上覆盖间隙：

# 修改线段生成逻辑，每条线段包含前一个线段的终点和当前的两个点
segments = []
for i in range(len(points)-1):
    # 让当前线段的起点提前一个点，和前一条线段重叠
    start_idx = max(0, i-1)
    segments.append(points[start_idx:i+2])
segments = np.array(segments)

# 后续创建LineCollection和导出步骤和方法1一样

这种方法有点笨，但胜在有效，适合样式参数失效的极端场景。

方法3：完全平滑的渐变线条（适合颜色过渡需求） #

如果你更想要颜色完全连续过渡的效果，而不是分段上色，那可以用散点+基础线条的方式模拟：

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 10, 200)
y = np.sin(x)
# 用x值映射颜色，实现完全平滑的渐变
norm = plt.Normalize(x.min(), x.max())
colors = plt.cm.plasma(norm(x))

fig, ax = plt.subplots()
# 用散点画每个点，大小对应线宽，颜色渐变
ax.scatter(x, y, color=colors, s=120, zorder=2)
# 底层用细黑线连接所有点，保证线条的连续性
ax.plot(x, y, color='black', linewidth=1, zorder=1)

ax.autoscale()
plt.savefig('smooth_gradient_line.pdf', dpi=300)
plt.show()

这种方法的颜色过渡绝对平滑，但本质是密集的点，线宽均匀性不如LineCollection，适合对颜色平滑度要求极高的场景。

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