在matplotlib中添加额外的线性图[重复]答案

【问题标题】：Add extra linear plot in matplotlib [duplicate]在matplotlib中添加额外的线性图[重复]
【发布时间】：2017-02-12 21:29:45
【问题描述】：

我有这个代码：

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()

上面的代码创建了这个图：

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

【问题讨论】：

您可能需要添加MWE，因为上面的代码需要定义data2。

标签： python pandas numpy matplotlib

【解决方案1】：

这是另一条蓝色线：

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

【讨论】：

【解决方案2】：

您可以将新图添加到相同的轴并设置它们的颜色（请参阅documentation 中的其他颜色选项）：

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

【讨论】：