如何用数值向量为Matplotlib双对数图添加渐变色彩？

plt.loglog（包括基础的plt.plot）的color参数仅支持单一颜色值，无法直接传入色彩向量实现线条的连续渐变。要达成这个需求，最直接的方案是使用matplotlib.collections.LineCollection，它可以将整条折线拆分为多个小线段，为每个线段分配对应颜色，再结合对数刻度和颜色归一化实现目标效果。

以下是完整的实现代码，和你提供的scatter示例数据保持一致：

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

# 准备数据
y = np.empty([40, 2])
y[:, 0] = np.arange(40)
y[:, 1] = y[:, 0]**2
color = y[:, 1]**3

# 创建线段集合：将连续点拆分为两两一组的小线段
segments = np.array([y[i:i+2, :] for i in range(len(y)-1)])

# 设置画布和坐标轴
fig, ax = plt.subplots()

# 创建LineCollection，指定颜色、色卡和对数归一化
lc = LineCollection(segments, cmap='OrRd', norm=mpl.colors.LogNorm(vmin=color.min(), vmax=color.max()))
lc.set_array(color[:-1])  # 每个线段对应一个颜色值，数量比点少1
lc.set_linewidth(2)  # 设置线条粗细

# 添加到坐标轴并设置对数刻度
ax.add_collection(lc)
ax.set_xscale('log')
ax.set_yscale('log')

# 自动调整坐标轴范围以适配数据
ax.autoscale()

# 添加颜色条
fig.colorbar(lc)

plt.show()

关键部分说明 #

• 线段拆分：segments将原始的点数组转化为多个小线段的集合，每个线段对应一对相邻点，这样才能为每个线段单独设置颜色。
• 颜色映射：通过norm=mpl.colors.LogNorm保持和你scatter示例一致的对数颜色归一化，确保色彩分布和原示例匹配。
• 颜色数组匹配：lc.set_array(color[:-1])因为线段数量比点少1，所以取color数组的前n-1个值对应每个线段的颜色。

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