python - 如何使用颜色图绘制 matplotlib 线图？

Question

如何使用 python matplotlib 创建线图，使得线条的颜色相对于另一个系列有所不同？

举个简单的例子:

import numpy as np
import matplotlib.pyplot as plt

t = np.arange(0.0, 2.0, 0.01)
y = np.sin(2 * np.pi * t)
z = (t-1) ** 2

fig = plt.figure()
ax = plt.axes()
ax.plot(t, y)
ax.plot(t, z)
plt.show()

而不是:

我只想以线条颜色表示遵循特定颜色图 (cmap)(例如“plasma”)的 z 值的方式绘制 (t, y) 图表。

编辑:

此问题被标记为可能重复，但引用了 question其中所需的结果是一条改变颜色的线，以帮助遵循它所绘制的路径(序列信息)，而不是添加另一个值的信息(在本例中为 z)。

更接近this example ，正如评论中所指出的，但我正在寻找比创建一组线段并单独为它们着色更简单的东西。

Answer 1

使用colorline函数引用in an answer to the question you said is not a duplicate of this ，可以这样做:

from colorline import colorline

colorline(t, y, z)
plt.xlim(t.min(), t.max())
plt.ylim(y.min(), y.max())
plt.show()

这会创建多个线段，每个线段的颜色由 z 确定。