如何在gnuplot中为2D绘图线条按x相关函数实现红蓝渐变着色

嘿，这个需求挺实用的！要实现f(x)的线条随g(x)从红到蓝平滑变色，核心就是把g(x)的数值和颜色空间做映射，再分段绘制线条。下面给你拆解思路和具体实现：

• 归一化g(x)数值：先把g(x)的[0,10]范围映射到[0,1]区间，这样能直接对应颜色渐变的参数。公式很简单：norm_g = g(x) / 10——当g(x)=0时norm_g=0，g(x)=10时norm_g=1，中间值线性过渡。
• 构建红-蓝渐变颜色：红色的RGB值是(1, 0, 0)，蓝色是(0, 0, 1)。我们可以对这两个颜色做线性插值，得到中间色：color = (1 - norm_g, 0, norm_g)。这样norm_g越小越红，越大越蓝，完全符合你的要求。
• 分段绘制线条：普通的单条plot命令没法给同一条线加渐变颜色，所以我们需要把x轴拆成很多细小的分段，给每个分段分配对应的颜色，最后拼接起来就成了平滑的渐变线条。

假设我们用f(x)=sin(x)、g(x)=x（x范围0到10）来演示，代码如下：

import matplotlib.pyplot as plt
import numpy as np

# 替换成你自己的f(x)和g(x)
def f(x):
    return np.sin(x)

def g(x):
    return x

# 生成密集的x点，保证渐变足够平滑
x = np.linspace(0, 10, 1000)
y = f(x)
g_values = g(x)

# 归一化g(x)到[0,1]区间
normalized_g = g_values / 10

# 创建画布和坐标轴
fig, ax = plt.subplots(figsize=(8, 4))

# 遍历每个小分段，逐个绘制带颜色的线段
for i in range(len(x) - 1):
    # 当前分段的x、y坐标
    x_segment = x[i:i+2]
    y_segment = y[i:i+2]
    # 计算当前分段的颜色
    current_color = (1 - normalized_g[i], 0, normalized_g[i])
    ax.plot(x_segment, y_segment, color=current_color, linewidth=2)

# 添加颜色条，方便查看g(x)和颜色的对应关系
color_mapper = plt.cm.ScalarMappable(cmap=plt.cm.RdBu, norm=plt.Normalize(vmin=0, vmax=10))
color_mapper.set_array([])
fig.colorbar(color_mapper, ax=ax, label="g(x) Value")

# 设置图表标签和标题
ax.set_xlabel("x")
ax.set_ylabel("f(x)")
ax.set_title("f(x) with Color Gradient Based on g(x)")

plt.tight_layout()
plt.show()

• 如果你的g(x)实际范围不是刚好0-10，只需要调整归一化公式：normalized_g = (g(x) - g_min) / (g_max - g_min)，其中g_min是g(x)的最小值，g_max是最大值，这样不管g(x)是什么范围都能完美映射到颜色渐变区间。
• 要是用Plotly这类交互式绘图库，实现会更简单：直接用scatter模式，设置mode='lines'，把color参数设为g(x)的数值，再指定colorscale='RdBu_r'（反转RdBu的默认蓝红顺序，变成红蓝渐变）就行，不用手动分段绘制。

内容的提问来源于stack exchange，提问作者no.clue