Matplotlib 2D线图，颜色作为第三个变量的函数，加上colorbar

Question

我试图绘制两个颜色的二维线图作为第三个数组的函数。 我的问题与此类似：matplotlib: varying color of line to capture natural time parameterization in data 我已经成功尝试了这种方法，下面是我当前代码的一个版本：

import matplotlib.pyplot as plt
import matplotlib.colors as clr
import matplotlib.collections as col
import numpy as np

def XYPlot(A):
    """
    Plots XY Plot of given trajectory, with color as a function of p
    """
    x = A[0][A[7] != 0]
    y = A[1][A[7] != 0]
    p = A[7][A[7] != 0]

    points = np.array([x,y]).T.reshape(-1,1,2)
    segments = np.concatenate([points[:-1], points[1:]], axis=1)

    lc = col.LineCollection(segments, cmap=plt.get_cmap('RdYlBu'),
    norm=plt.Normalize(200, 1000))
    lc.set_array(p)
    lc.set_linewidth(1.5)

    plt.gca().add_collection(lc)
    plt.xlim(-25,25)
    plt.ylim(-25,25)

    plt.show()
    pass

但是，我想知道是否有更方便的方法，而不是使用LineCollection。

无论如何，我需要在我的情节中添加第三个变量的颜色条。我该如何创建一个？

谢谢！