根据参数绘制具有不同颜色的曲线[重复]

我想用一些测量数据绘制一条曲线。它可以作为包含以下形式的项目的数组提供[t,b]，其中 t 是我要绘制的参数，b是描述测量设备状态的字符串。我现在想绘制的值t并根据值对线条进行着色b。到目前为止我的代码是

import pylab as pl
measurements = [[0, "a"], [1, "b"], [2, "c"]]
times = pl.arange(0, 3, 1)
values = zip(*measurements)[0]
parameters = zip(*measurements)[1]
pl.plot(times, values)
pl.show()

该线现在应该具有不同的颜色，具体取决于中的值parameters。我该怎么做呢？

最好的方法是使用 matplotlibcolormap API.

from matplotlib import pyplot as pl
from matplotlib import cm

pl.scatter(times, values, c = cm.spectral(1.*values/max(values)))

显然，您需要使用其他函数来从数据中获取颜色。但总的来说，cm.spectral () 将返回给定 0 到 1 之间的浮点数或 0 到 255 之间的 int 的颜色。

作为一般性评论，您可能最好使用 numpy 数组，而不是尝试zip您的数据在一起。对于更复杂的数据结构可能比这个例子更容易。

import numpy as np
from matplotlib import pyplot as pl
from matplotlib import cm

measurements = np.array([[0, "a"], [1, "b"], [2, "c"]])
times = np.arange(3)
values = np.array(measurements[:,0], dtype=float)
parameters = np.array(measurements[:,1], dtype='S1')

pl.scatter(times, values, c = cm.spectral(values/max(values)))
pl.show()