不同颜色的matplotlib线取决于其他变量

我认为colormap对于连续数据很有用。对于离散颜色列表,最好使用离散颜色列表。因此,可以将颜色与类型变量配对:

代码:

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors as colors


income = np.array([5000, 5005, 5010, 6000, 6060, 6120, 7000])
x = np.arange(len(y))
job = np.array([0, 0, 0, 1, 1, 1, 2]).astype('int')

# iterate over zipped job and color
for j, c in zip(job, colors.TABLEAU_COLORS):
    plt.plot(x[job == j], income[job == j], 'o-', c=c)

plt.show()

绘图:

TABLEAU_COLORS 但是你可以找到另一个颜色列表 here 如果你愿意的话。