添加额外的线性图中matplotlib

Question

我有这样的代码：

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

上面的代码创建了这个阴谋：

我想添加一个名为多个线性函数“预测2”蓝色/绿线的颜色。我该怎么做呢？

Answer 1

您可以添加新的地块以同一轴线，并设置自己的颜色（见documentation其他颜色可供选择）：

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

Answer 2

下面是蓝色的另一条线：

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