如何在同一个图中绘制两个列表，但使用不同的颜色？

Question

请帮我在同一张图上绘制两个列表。线应该是不同的颜色。以下是我试过的代码：

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

您应该使用plt.plot两次来绘制两行。

我不知道你的X轴是什么，但显然你应该创建另一个数组/列表作为你的X值。

然后使用plt.plot(x_value,forecast, c='color-you-want') 和plt.plot(x_value,train_z, c='another-color-you-want')。

。有关更多详细信息，请参阅pyplot文档。

Answer 2

窃取另一个答案借款，这似乎工作：

# 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() 

对方回答：python/matplotlib - multicolor line

当然，与任何更换'r'，'g'，'b'值其他https://matplotlib.org/api/pyplot_api.html#matplotlib.pyplot.plot颜色列表