python画图一条线两种颜色_Python：如何以不同的颜色绘制一条线

LEATH

从此示例复制：import numpy as npimport matplotlib.pyplot as pltfrom matplotlib.collections import LineCollectionfrom matplotlib.colors import ListedColormap, BoundaryNormx = np.linspace(0, 3 * np.pi, 500)y = np.sin(x)z = np.cos(0.5 * (x[:-1] + x[1:]))  # first derivative# Create a colormap for red, green and blue and a norm to color# f' < -0.5 red, f' > 0.5 blue, and the rest greencmap = ListedColormap(['r', 'g', 'b'])norm = BoundaryNorm([-1, -0.5, 0.5, 1], cmap.N)# Create a set of line segments so that we can color them individually# This creates the points as a N x 1 x 2 array so that we can stack points# together easily to get the segments. The segments array for line collection# needs to be numlines x points per line x 2 (x and y)points = np.array([x, y]).T.reshape(-1, 1, 2)segments = np.concatenate([points[:-1], points[1:]], axis=1)# Create the line collection object, setting the colormapping parameters.# Have to set the actual values used for colormapping separately.lc = LineCollection(segments, cmap=cmap, norm=norm)lc.set_array(z)lc.set_linewidth(3)fig1 = plt.figure()plt.gca().add_collection(lc)plt.xlim(x.min(), x.max())plt.ylim(-1.1, 1.1)plt.show()