Python：如何在twinx之后制作传奇

Question

fig, ax1 = plt.subplots()
ax2 = ax1.twinx()

ax1.plot(x,y1,'go-')
ax1.plot(x,y2,'ro-')
ax1.plot(x,y3,'g*--')
ax1.plot(x,y4,'r*--')
ax2.plot(x,y5,'bo-')
ax2.plot(x,y6,'b*--')

ax1.set_xlabel('cluster size')
ax1.set_ylabel('number of nodes', color='g')
ax2.set_ylabel('error rate', color='b')

我想绘制如上所述的6条曲线。有两个Y轴。 我不知道如何为这6条曲线制作传奇。 我想得到的就像

legend([y1,y2,y3,y4],['A','B','C','D'])
legend([y5,Y6],['e','f'])

Answer 1

您可以为每个plot()语句使用标签。例如：

ax1.plot(x,y1,'go-', label='A')
ax1.plot(x,y2,'ro-', label='B')
ax1.plot(x,y3,'g*--', label='C')
ax1.plot(x,y4,'r*--', label='D')

然后，拨打

ax1.legend()

会自动创建您想要的图例，因为在没有参数的情况下调用legend()会自动获取处理程序及其各自的标签。

如果您想要单独的图例，您还可以通过loc参数为它们指定不同的位置。查看guide了解详情。