如何以不同的颜色绘制一条线以继续另一个情节？ Matplotlib

Question

I have a question about plotting data in Matplotlib. How can I plot a line in a different colour to continue another line in Matplotlib (as shown below) to show a split in Training/Testing data or Current Price/Predicted Price?

I am using Python3 in Jupyter Notebook if it helps. Thanks!

Answer 1

You can split the training data and testing data into two and plot them in different colour in the same figure.

plt.plot(train_x, train_y, 'blue', test_x, test_y, 'orange')
plt.legend(["Train","Test"])
plt.show()