1、首先在绘图时，加入label标签

plt.plot(x, y1, label='自己', color = 'pink', linestyle = '-.')
plt.plot(x, y2, label='同桌', color = 'cyan', linestyle = ':')

2、设置显示图例函数

#添加图例  loc位置
plt.legend()

loc表示位置，在函数定义中的值分别是：

'best' : 0

'upper right' : 1

'upper left' : 2

'lower left' : 3

'lower right' : 4

'right' : 5

'center left' : 6

'center right' :７

'lower center' : 8

'upper center' : 9

'center' : 10