Python-Legend图例

import matplotlib.pyplot as plt  #matplotlib.pyplot缩减为plt
import numpy as np    #数据的库

x=np.linspace(-3,3,50) #-1到1的50个点
y1=2*x+1  #一次函数
y2=x**2+1  #二次函数
plt.figure()

plt.xlim((-1,2))
plt.ylim((-2,3))
plt.xlabel('I am x')
plt.ylabel('I am y')

new_ticks=np.linspace(-1,2,5) #设置新的小标
plt.xticks(new_ticks)#将x的刻度换成新的小标
plt.yticks([-2,-1.8,-1,1.22,3,],[r'$really\ bad$','$bad\ \\alpha $','$normal$','$good$','$really\ good$'])

l1,=plt.plot(x,y1,label='up') #绘制出来
l2,=plt.plot(x,y2,color='red',linewidth=1.0,linestyle='--',label='down') #绘制出来
#plt.legend() #加图例
plt.legend(handles=[l1,l2,],labels=['aaa','bbb'],loc='best')
#plt.legend(handles=[l1,],labels=['aaa',],loc='best') #只有aaa的图例
plt.show() #显示出来