如何用不同图例的彩虹色函数for loop绘制图形

Question

我想用python绘制3个数字。我手工绘制的。
代码是：

import matplotlib.pyplot as plt
import numpy as np    
fig, ax = plt.subplots(1, 1)    
Semesters=np.arange(1,7,1)
y1=np.arange(10,16,1)
y2=np.arange(20,26,1)
y3=np.arange(51,57,1)
plt.plot(Semesters, y1,label="To Reach 120",linewidth=2)
plt.plot(Semesters, y2,label="To Reach 100",linewidth=2)
plt.plot(Semesters, y3,label="To Reach 80",linewidth=2)
ax.set_xticks(Semesters)
plt.legend(bbox_to_anchor=(0.85, .3), loc=2, borderaxespad=0.)

我想用for loop和rainbow函数（或任何其他顺序颜色函数）来考虑图例的颜色。感谢任何帮助。在

Answer 1

这里有一个解决方案。在

解释：在列表中创建y数据（y_list此处）并存储标签（labels此处）。然后按照您的要求，使用for循环遍历列表，一次绘制一个。在

例如，要使用颜色映射来定义颜色，rainbow可以使用线（图）的数量创建一个颜色列表（在下面的示例中为3）。然后只要在for循环中指定这些颜色，就可以得到所需的绘图。在

import matplotlib.pyplot as plt
import numpy as np    
from matplotlib import cm

fig, ax = plt.subplots(1, 1)    
Semesters=np.arange(1,7,1)
y_list = [np.arange(10,16,1), np.arange(20,26,1), np.arange(51,57,1)]
labels = ["To Reach 120", "To Reach 100", "To Reach 80"]

# Define the colors to be used using rainbow map (or any other map)
colors = [cm.rainbow(i) for i in np.linspace(0, 1, len(y_list))]

# Plot the lines using a for loop
for i in range(len(y_list)):
    plt.plot(Semesters, y_list[i], label=labels[i], linewidth=2., color=colors[i])

ax.set_xticks(Semesters)
plt.legend(bbox_to_anchor=(0.85, .3), loc=2, borderaxespad=0.)