如何根据变量值动态更改Matplotlib绘图线条颜色？

plt.plot()的color参数只接受单个颜色字符串/颜色值，不能直接传入颜色列表，这就是你遇到类型错误的原因。要实现按s的值分段改变线条颜色，需要把数据分成连续的、s值相同的片段，逐个片段绘制并指定对应颜色。

import matplotlib.pyplot as plt
# 保留你的初始数据生成代码
x, s = hmm.rand(500)
plt.figure(figsize=(10,5))

# 将s转为一维数组（适配可能的二维输入）
s_flat = s.flatten()
# 找出s值发生变化的索引位置，添加首尾索引用于分割片段
change_indices = [0] + list((s_flat[1:] != s_flat[:-1]).nonzero()[0] + 1) + [len(s_flat)]

# 遍历每个连续片段，逐段绘制
for i in range(len(change_indices)-1):
    start = change_indices[i]
    end = change_indices[i+1]
    # 根据当前片段的s值确定颜色
    current_color = 'red' if s_flat[start] < 0.5 else 'blue'
    # 绘制对应区间的x数据
    plt.plot(range(start, end), x[0,0,start:end], color=current_color)

plt.xlabel("t")
plt.ylabel("X[t]")
plt.title("Example output over 500 states")
plt.show()

• 先把s转为一维数组s_flat，避免多维数组带来的处理麻烦；
• change_indices定位所有s值切换的位置，结合首尾索引把数据分割成连续的同值片段；
• 循环处理每个片段，根据该段的s值匹配颜色，再绘制对应区间的x数据，实现分段变色的线条效果。

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