python - 如何在 matplotlib 中使用单个 plot 命令绘制多色曲线

Question

如果我没理解错的话，函数:

matplotlib.pyplot.plot(x, y)

绘制 len(x)-1 个单独的线段 - 一个从 (x[0], y[0]) 到 (x[1],y[1])，一个从 (x[1], y[1]) 到 (x[2], y[2]), etc. 在我的应用程序中，我想显示一条由一系列连接数据点的线段组成的曲线，但是多了一条与每个数据点之间的转换相关联的数据 (z)，我想用线段的颜色表示。显然，一种方法如下:

for i in range(len(x)-1)):
    matplotlib.pyplot.plot(x[i:i+2],y[i:i+2], color=z[i])

但是有没有一种方法可以做到这一点而不需要为每个线段单独调用 matplotlib.pyplot.plot？

Answer 1

您应该能够使用 matplotlib.collections.LineCollection它接受一个 colors 参数(RGBA 元组序列)。