Pyplot：根据类别改变颜色

Question

我有一组值及其关联的类标签（0或1）。我想根据类标签更改绘制值的颜色。 我正在使用matplotlib.pyplot绘图函数来绘制值：

plt.plot(data[0])

对于每个值，关联的类标签存储在与数据数组长度相同的单独数组中。

目前的情节如下：

红线之间的区域应采用不同的颜色。

Answer 1

您可以将其拆分为两个不同的数据集：

xx0 = class_labels == 0
xx1 = class_labels == 1

data_class_0 = data[0].copy()
data_class_0[xx1] = np.nan

data_class_1 = data[0].copy()
data_class_1[xx0] = np.nan

plt.plot(data_class_0, 'b')
plt.plot(data_class_1, 'r')