如何更改子图中不同行的线条颜色

Question

我希望能够使用matplotlib将线条的颜色更改为我在绘图上指定的颜色。

以下是我正在使用的相关代码（它是函数的一部分）。变量avg_rel_track是'2D数组'，每列对应一个'刀片'。每个“刀片”都绘制一条单独的线。

我希望每个刀片/线都是我指定的颜色。我很难找到相关的文档，如果这很明显，我很抱歉。

 def plot_data(avg_rel_track, sd_rel_track_sum, shade):

    fig = plt.figure(figsize=(18,10))

    gs  = gridspec.GridSpec(5, 1, height_ratios=[1.75, 1 ,1, 1])
    ax0 = plt.subplot(gs[0])
    ax1 = plt.subplot(gs[1])
    ax2 = plt.subplot(gs[2])
    ax3 = plt.subplot(gs[3])
    ax4 = plt.subplot(gs[4])

    fig.subplots_adjust(top=0.93)

    #The following plot has 5 plots within it.
    lineObjects = ax0.plot(avg_rel_track_nan)
    ax0.set_title('Averaged Relative Track',fontsize=11)
    ax0.legend(lineObjects, (1,2,3,4,5),loc='lower center', bbox_to_anchor=(0.82, 1),
      fancybox=True, shadow=True, ncol=5)

以下是情节的一个例子（非常粗略）我想将情节改为我指定的情节。它是5个子图中的一个作物

Answer 1

问题是plot函数只接受一种颜色作为参数。因此，您必须迭代列并分别绘制每个列。

columns = [[1,2,3],[1,4,5],[6,4,2]]
colors = ['green', 'pink', 'blue']
labels = ['foo', 'bar', 'baz']

fig, ax = plt.subplots(1, 1)
for i, column in enumerate(columns):
    ax.plot(column, color=colors[i], label=labels[i])
ax.legend(loc='lower center', bbox_to_anchor=(0.82, 1),
          fancybox=True, shadow=True, ncol=3)