在matplotlib中添加额外的线性图

时间:2016-10-04 16:03:27

标签: python pandas numpy matplotlib

我有这段代码:

x = np.linspace(data2.X.min(), data2.X.max(), 100) 
f = 0.1 + (0.2 * x)
fig, ax = plt.subplots(figsize=(12,8))
ax.plot(x, f, 'r', label='Prediction #1')
ax.scatter(data2.X, data2.Y, label='Raw Data') 
ax.legend(loc=2)  
ax.set_xlabel('X') 
ax.set_ylabel('Y')
plt.show()

上面的代码创建了这个图: Plot

我想添加一个名为“Prediction#2”的线性函数,其中蓝色/绿色线条颜色。我该怎么做?

2 个答案:

答案 0 :(得分:1)

您可以将新图添加到相同的轴并设置其颜色(请参阅documentation中的其他颜色选项):

...
ax.plot(blue_x, blue_f, 'b', label='Prediction Blue')
ax.plot(green_x, green_f, 'g', label='Prediction Green')

答案 1 :(得分:0)

这是蓝色的另一条线:

 $('#snitch_<%= @snitch.token %>_icon').append("")