Pyplot: Ändern der Farbe in Abhängigkeit von der Klasse

Question

Ich habe ein Array von Werten mit ihren zugeordneten Klassenbeschriftungen (0 oder 1). Ich möchte die Farbe der gezeichneten Werte basierend auf ihren Klassenbeschriftungen ändern. Ich verwende die matplotlib.pyplot Plotfunktion, um die Wert zu Parzelle:

plt.plot(data[0]) 

für die zugeordneten einzelnen Wert Klasse Etiketten in einer separaten Anordnung von der gleichen Länge wie die Daten-Array gespeichert sind.

Der aktuelle Plot sieht wie folgt aus:

Die Bereiche zwischen den roten Linien unterschiedlich gefärbt werden soll.

Answer 1

Man konnte es in zwei verschiedenen Datensätzen aufgeteilt:

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