Plotting list of lists in a same graph in Python

Question

I am trying to plot (x,y) where as y = [[1,2,3],[4,5,6],[7,8,9]].

Say, len(x) = len(y[1]) = len(y[2]).. The length of the y is decided by the User input. I want to plot multiple plots of y in the same graph i.e, (x, y[1],y[2],y[3],...). When I tried using loop it says dimension error.

I also tried: plt.plot(x,y[i] for i in range(1,len(y)))

How do I plot ? Please help.

for i in range(1,len(y)):
    plt.plot(x,y[i],label = 'id %s'%i)
    plt.legend()
    plt.show()

You have an indentation error in the plot command. You are also starting the indexing from 1. {range (1,...) instead of range(0,...) ). I am assuming that is intentional. The code should work nonetheless. Kindly let us know the error you are facing — R. S. Nikhil Krishna
– R. S. Nikhil Krishna, Commented Oct 16, 2016 at 17:42
X=[1,2,3]. @Nikhil I am getting dimension error while plotting — sivasudhan
– sivasudhan, Commented Oct 16, 2016 at 18:03
I also tried ax=plt.subplot (111) and yet I get different plots for different Y's — sivasudhan
– sivasudhan, Commented Oct 16, 2016 at 18:05

PIZZA PIZZA · Accepted Answer · 2019-06-04 21:59:36Z

28

Assuming some sample values for x, below is the code that could give you the desired output.

import matplotlib.pyplot as plt
x = [1,2,3]
y = [[1,2,3],[4,5,6],[7,8,9]]
plt.xlabel("X-axis")
plt.ylabel("Y-axis")
plt.title("A test graph")
for i in range(len(y[0])):
    plt.plot(x,[pt[i] for pt in y],label = 'id %s'%i)
plt.legend()
plt.show()

Assumptions: x and any element in y are of the same length. The idea is reading element by element so as to construct the list (x,y[0]'s), (x,y[1]'s) and (x,y[n]'s.

Edited: Adapt the code if y contains more lists.

Below is the plot I get for this case:

edited Jun 4, 2019 at 21:59

PIZZA PIZZA

5932 gold badges9 silver badges21 bronze badges

answered Oct 16, 2016 at 18:03

Sreejith Menon

1,0873 gold badges18 silver badges27 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

PIZZA PIZZA Over a year ago

One improvement that would adapt y values with more than 3 lists (e.g y = [[1,2,3],[4,5,6],[7,8,9],[10,11,12]] ), is to set the range in the for loop as "for i in range(len(y[0])):"

Aizzaac Over a year ago

When i execute your code i get: ‘list’ object has no attribute ‘tolist’

Nathan Pyle · Accepted Answer · 2019-04-07 05:51:29Z

10

Use a for loop to generate the plots and use the .show() method after the for loop.

 import matplotlib.pyplot as plt
 for impacts in impactData:
     timefilteredForce = plt.plot(impacts)
     timefilteredForce = plt.xlabel('points')
     timefilteredForce = plt.ylabel('Force')

 plt.show()

impactData is a list of lists.

edited Apr 7, 2019 at 5:51

answered Aug 28, 2018 at 23:20

Nathan Pyle

6217 silver badges11 bronze badges

2 Comments

Gabriel Fair Over a year ago

Thanks for this code! A minor point of correction, its pyplt.plot(impacts not pyplt.plt. cheers!

Nathan Pyle Over a year ago

Thanks for catching that. I updated the import to match the standard plt call.

Collectives™ on Stack Overflow

Plotting list of lists in a same graph in Python

2 Answers 2

2 Comments

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

2 Comments

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related