python - Matplotlib 图例显示了一些意想不到的代码线

Question

我正在学习 matplotlib 库，因此尝试在 1 张图上显示 2 条线来表示机器学习模型。问题在于传说。图例中有一些代码行。我不知道如何删除它。有人可以帮忙删除它吗？

alldata=plt.plot(x,y,'o')
l1 =plt.plot(x_new, model1(x_new),'r',linestyle='dashed',linewidth=4)
l2 =plt.plot(x_new, model2(x_new),'g',linewidth=4)
plt.title('no. of bench employees needed for new projects over last 5 years')
plt.xlabel('year')
plt.ylabel('employees/month')
plt.xticks([W*12 for W in range(6)],['year %i'%w for w in range(6)])
plt.legend([("alldata"),(l1,"d=%i" %model1.order),(l2,"d=%i" %model2.order)], loc=1)
plt.autoscale(tight=True)
plt.grid()
plt.show()

上面的代码生成此图:

Answer 1

如果你看一下matplotlib.pyplot.legend文档，您会注意到您使用了错误的格式调用它。引用文档:

For full control of which artists have a legend entry, it is possible to pass an iterable of legend artists followed by an iterable of legend labels respectively:

legend((line1, line2, line3), ('label1', 'label2', 'label3'))

因此您需要传入两个元组:

plt.legend((alldata[0], l1[0], l2[0]),                                # plots
           ("alldata","d=%i" % model1.order, "d=%i" % model2.order),  # names
           loc=1)