添加額外的線性圖中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()