2D array plotting in matplotlib

Question

I was trying to plot a 2D array vs a 1D array with pyplot. I can do that with no problems and columns in the 2D array are treated like two different sets of Y datas, which is what i want. What i don't know is how to specify a different color for every column in the 2d array. if i use pyplot.plot(1darray, 2darray, "r-") every column in 2d array is plotted red for example. Should i modify the standard color map or is there a smarter way?

Have you tried leaving out the color specification, and just let matplotlib choose the colors? I mean this: pyplot.plot(1darray, 2darray, "-") — Björn Pollex, Commented Mar 25, 2011 at 8:30
yes, but i want to choose the colors for the different columns. — purpleshift, Commented Mar 25, 2011 at 12:19

Björn Pollex · Accepted Answer · 2011-03-25 12:42:48Z

1

If you want to use custom colors for each column, then the best approach is to plot each column explicitly using a loop:

for column, colcolor in zip(2darray, colors):
    pyplot.plot(2darray, column, "-", color=colcolor)

You may have to use 2darray.T, I'm not sure, and I can't check right now.

answered Mar 25, 2011 at 12:42

Björn Pollex

76.9k29 gold badges205 silver badges289 bronze badges

1

no, normally you should transpose them but the cycle pick up the first element anyway
– purpleshift
Commented Mar 25, 2011 at 14:19

Add a comment |

Collectives™ on Stack Overflow

2D array plotting in matplotlib

1 Answer 1

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Related