Pyplot：クラスによって色が変わります

Question

私は関連するクラスラベル（0または1）で値の配列を持っています。クラスラベルに基づいてプロットされた値の色を変更したいと思います。 Iの値をプロットするmatplotlib.pyplotプロット関数を使用しています：関連付けられたクラスラベルはデータ配列と同じ長さの別の配列に格納されている各値について

plt.plot(data[0]) 

。

現在のプロットは、次のようになります。

赤線の間にある領域が異なる色に着色されなければなりません。

Answer 1

次の2つの異なるデータセットでそれを分割することができます：

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')