在matplotlib中绘制二维数组

Question

我在用pyplot画图的时候，想把一个二维数组和一个一维数组一起画出来。这个没问题，二维数组的每一列就像是两组不同的Y数据，这正是我想要的。不过，我不知道怎么给二维数组的每一列指定不同的颜色。如果我用pyplot.plot(1darray, 2darray, "r-")，那么二维数组的每一列都会被画成红色。请问我是不是应该修改标准的颜色映射，还是有更聪明的方法呢？

Answer 1

如果你想为每一列使用自定义颜色，最好的方法就是通过一个循环来逐个绘制每一列：

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

你可能需要用到 2darray.T，我不太确定，现在也没法检查。