Pyplot/matplotlib线图 - 相同的颜色

Question

我用matplotlib/pyplot绘制一个简单的线图.在同一个图中,最多有20行.Matplotlib自动选择线条颜色.我需要用较粗的红色画出第一行.

所以我试过:

if (i==0):                       # first line
    plt.plot(x, y, 'r', lw=2.5)
else:                            # other lines
    plt.plot(x, y)

到目前为止非常简单.第一行是正确的.但其他一条线也是红色的.怎么能避免这个？

Answer 1

默认jet颜色映射从蓝色变为红色.每个内置色彩映射也都有一个姐妹色图,颜色的顺序相反.反转颜色映射与后缀具有相同的名称_r.因此,jet_r从红色到蓝色.

因此,您可以jet_r通过定义从色彩图中为每一行赋予唯一的颜色

color = cmap(float(i)/N)

这里cmap是jet_r颜色表.色图cmap是可调用的.传递一个浮点数时,它返回一个RGB元组.因此,如果float(i)/N是每行的0到1之间的明显浮点数,那么每一行将自动获得不同的颜色.

例如,

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 5, 100)
cmap = plt.get_cmap('jet_r')
N = 20
for i, n in enumerate(np.linspace(2.0, 0, N)):
    y = np.sin(x)*x**n
    color = cmap(float(i)/N)
    lw = 2.5 if i==0 else 1
    # zorder controls the order in which the lines are drawn. 
    # The line with the highest zorder lies on top.
    plt.plot(x, y, c=color, lw=lw, zorder=-i)
plt.show()

产量