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