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

Question

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

下面是我正在使用的相关代码(它是函数的一部分)。变量avg_rel_track是一个“二维数组”，每一列对应一个“ Blade ”。每个“ Blade ”绘制一条单独的线。

我希望每个 Blade /线都是我指定的颜色。我正在努力寻找相关文档，如果它很明显，我很抱歉。

 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)