图例

方法一

import matplotlib.pyplot as plt
import numpy as np

x=np.arange(1,10)
y=x+2
yy=2*x+1

fig=plt.figure()
ax=plt.gca()
ax.plot(x,y,label='a',c='coral')
ax.plot(x,yy,label='b',c='y')
plt.legend()
plt.show()

在plot中的参数label注明标注名称即可实现图例的绘制，但不要忘了plt.legend()让图例显示在图中。

方法二

import matplotlib.pyplot as plt
import numpy as np


x=np.arange(1,10)
y=x+2
yy=2*x+1

fig=plt.figure()
ax=plt.gca()
l1,=ax.plot(x,y,c='coral')
l2,=ax.plot(x,yy,c='y')
plt.legend(handles=[l1,l2],labels=['aaa','bbb'],loc='best')

plt.show()

这种方法是在legend中设置图例，首先需要给所画的图起一个名字即l1，l2,但不要忘了后边还有一个逗号，原因是返回值有多个这里只对第一个返回值起名字。然后像上面一样在legend中设置即可。