How to plot two list on the same graph with different colors?

[Alberto]

Dear Python Users,

I tried to plot a graph from two lists on the same graph. However, what I get is that two lists are plotted against each other. How can I plot two lists on the same graph, but with different colors?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19

import matplotlib.pyplot as plt   train_W = [1,2,3,4,5,6] 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()

[MTVDNA]

If you submit two arrays as arguments, it is assumed that they are x-coordinates and y-coordinates respectively. If you use only one array, these values will be used as y-coordinates and the x-coordinates will be [0, 1, 2, ...].

You have multiple options to plot more than one function.

You can submit x and y values for each graph you want to plot:

1 2	x_list = range(6) plt.plot(x_list, forecast, x_list, train_W, x_list, train_Z)

The colors of the lines will be different by default, but if you want to set them manually you could do that like this:

1	plt.plot(x_list, forecast, 'r-.', x_list, train_W, 'bx', x_list, train_Z, 'y')

But imho this code is not very easy to read. (note that 'r-.' means a dashed red line, 'bx' means blue 'x' markers with no line)

Note that the length of the arrays must be equal, or you'll run into e.g.:

Error:
ValueError: x and y must have same first dimension, but have shapes (6,) and (5,)

Also, if you want to apply other keyword arguments to the plot they'll apply to each of those lines:

1 2 3

plt.plot(x_list, forecast, x_list, train_W, x_list, train_Z, label='test') plt.legend() plt.show()

Each line in this plot will have the same entry in the in the legend.


That's why I prefer method 2: calling plot() separately for each line:

1 2 3

plt.plot(forecast) plt.plot(train_Z) plt.show()

or with x-values explicitly included:

1 2 3

plt.plot(x_list, forecast) plt.plot(x_list, train_Z) plt.show()

or with different keyword arguments for each line

1 2 3 4

plt.plot(x_list, forecast, label='forecast') plt.plot(x_list, train_Z, label='train Z') plt.legend() plt.show()

You can find more info about how to use arguments for plot here: https://matplotlib.org/api/pyplot_api.ht...yplot.plot

[Alberto]

Thank you!