Plotting a list of arrays on the same plot in python?

Question

I have a list of arrays or an array of arrays that looks like

a=[array1([.....]),array2([.....]),array3([....]),.....]

and a separate array b (not a list)

b=np.array[()]

All the arrays in the list "a" are the same length and the same length as "b". I want to plot all the arrays in list "a" on the y axis verses "b" on the x axis all on the same plot. So one plot that consists of a[0]vs b, a[1] vs b, a[2] vs b,...and so on.

How can I do this?

I tried

f, axes = plt.subplots(len(a),1)
for g in range(len(a)):
    axes[g].plot(b,a[g])
    plt.show()

but this gives me many plots stacked on each other and they don't even have all the data. I want everything on one plot.

Community · Accepted Answer · 2017-05-23 12:02:20Z

1

I just found some old code that should accomplish this. Try:

import random
import matplotlib.pyplot as plt
from matplotlib import cm
import numpy as np

# define a and b here

# Helps pick a random color for each plot, used for readability
rand = lambda: random.randint(0, 255)
fig = plt.figure(figsize=(10,7.5))
ax = fig.add_subplot(111)
for ydata in a:
    clr = '#%02X%02X%02X' % (rand(),rand(),rand())
    plot, = ax.plot(b, ydata, color=clr)

Edit: To generate the same set of colors every time, as answered in this post, try:

colors = cm.rainbow(np.linspace(0, 1, len(a)))
for ydata, clr in zip(a, colors):
    plot, = ax.plot(b, ydata, color=clr)

np.linspace gives you "evenly spaced numbers over a specified interval", [0,1] for this purpose.

edited May 23, 2017 at 12:02

CommunityBot

11 silver badge

answered Feb 17, 2017 at 23:18

sgrg

1,2509 silver badges15 bronze badges

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

4 Comments

Nabeel Eh Over a year ago

Thanks this worked great. Is there a way to make it so that the colors are always the same? So all the plots have different colors, but every time I plot, they stay the same color.

sgrg Over a year ago

Yup it's definitely possible. I haven't tried this but see edits.

Nabeel Eh Over a year ago

Thanks!! You're Awesome!

sgrg Over a year ago

If your issue is resolved, please mark an answer as accepted :)

Collectives™ on Stack Overflow

Plotting a list of arrays on the same plot in python?

1 Answer 1

4 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

4 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related