如何在matplotlib中创建图例

时间:2015-05-29 14:06:42

标签: python matplotlib

我正在研究回归问题,我想绘制3个DataFrame。我不知道如何设置Dataframes的标签。我想要blue-> ACTUAL,green-> SVR,red-> MLR。

代码有什么问题?

ax1 = y_test[1800:1900].plot(color='blue', linewidth=3)
predicted_y[1800:1900].plot(color='green', linewidth=3, ax =ax1)
predicted_y1[1800:1900].plot(color='red', linewidth=3, ax=ax1)
plt.legend(loc='upper center', bbox_to_anchor=(0.5, 1.05),   prop={'size':35})
plt.show()

我绘制了这个,它显示了所有带有0值的颜色。

1 个答案:

答案 0 :(得分:0)

我认为如果您为地块添加标签应该会有效:

ax1 = y_test[1800:1900].plot(color='blue', linewidth=3, label = 'ACTUAL')
predicted_y[1800:1900].plot(color='green', linewidth=3, ax =ax1, label = 'SVR')
predicted_y1[1800:1900].plot(color='red', linewidth=3, ax=ax1, label = 'MVR')
plt.legend(loc='upper center', bbox_to_anchor=(0.5, 1.05),   prop={'size':35})
plt.show()