Wie zeichne ich zwei Listen in der gleichen Grafik, aber mit verschiedenen Farben?

Question

Bitte helfen Sie mir, zwei Listen auf dem gleichen Graphen zu plotten. Die Linien sollten verschiedene Farben haben. Hier ist der Code, den ich versucht:

import matplotlib.pyplot as plt 
train_X = [1,2,3,4,5] 
train_Y = [10, 20, 30, 40, 50] 
train_Z = [10, 20, 30, 40, 50,25] 
alpha = float(input("Input alpha: ")) 
forecast = [] for x in range(0, len(train_X)+1): 
    if x==0:  
     forecast.append(train_Y[0]) 
    else: 
     forecast.append(alpha*train_Y[x-1] + (1 - alpha) * forecast[x-1]) 
plt.plot(forecast,train_Z,'g') 
plt.show() 

Answer 1

Sie plt.plot zweimal verwenden sollten zwei Linien zu zeichnen.

Ich weiß nicht, was ist Ihre X-Achse, aber offensichtlich sollten Sie ein anderes Array/Liste zu Ihrem X-Wert erstellen.

Dann verwenden Sie plt.plot(x_value,forecast, c='color-you-want') und plt.plot(x_value,train_z, c='another-color-you-want').

. Weitere Informationen finden Sie in der Dokumentation pyplot.

Answer 2

Stehlen Borrowing aus einer anderen Antwort, scheint dies zu funktionieren:

# plt.plot(forecast,train_Z,'g') # replace this line, with the following for loop 

for x1, x2, y1,y2 in zip(forecast, forecast[1:], train_Z, train_Z[1:]): 
    if y1 > y2: 
     plt.plot([x1, x2], [y1, y2], 'r') 
    elif y1 < y2: 
     plt.plot([x1, x2], [y1, y2], 'g') 
    else: 
     plt.plot([x1, x2], [y1, y2], 'b') # only visible if slope is zero 

plt.show() 

andere Antwort: python/matplotlib - multicolor line

Natürlich ersetzen die 'r', 'g', 'b' Werte mit jedem andere in https://matplotlib.org/api/pyplot_api.html#matplotlib.pyplot.plot Farbenliste