如何在 Matplotlib 图中的一行中获得不同的颜色？

Question

我正在使用 matplotlib 创建绘图。我必须在图表中画一条线，必须在每个点的函数中定义颜色。例如，我需要一条线，将 2000 以下的点涂成红色，将 2000 以上的点涂成蓝色。我怎样才能得到这个？你知道实现它的类似解决方案或解决方法吗？

这是我的示例代码，将孔线涂成蓝色（我猜是默认颜色）

def draw_curve(points, labels):     

    plt.figure(figsize=(12, 4), dpi=200)

    plt.plot(labels,points)

    filename = "filename.png"

    plt.savefig("tmp/{0}".format(filename)) 

    figure = plt.figure()

    plt.close(figure)

因此，在下图中，我希望将浅蓝色水平线上方的值绘制成与下方值不同的颜色。

提前致谢。

Answer 1

你必须为你的线的每一段着色：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.collections import LineCollection
from matplotlib.colors import ListedColormap, BoundaryNorm

# my func
x = np.linspace(0, 2 * np.pi, 100)
y = 3000 * np.sin(x)

# select how to color
cmap = ListedColormap(['r','b'])
norm = BoundaryNorm([2000,], cmap.N)

# get segments
xy = np.array([x, y]).T.reshape(-1, 1, 2)
segments = np.hstack([xy[:-1], xy[1:]])

# make line collection
lc = LineCollection(segments, cmap = cmap, norm = norm)
lc.set_array(y)

# plot
fig, ax = plt.subplots()
ax.add_collection(lc)
ax.autoscale()
plt.show()

更多示例：http://matplotlib.org/examples/pylab_examples/multicolored_line.html