P5-Legend图例
import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(-3, 3, 50)
y1 = 2*x + 1
y2 = x**2
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)
print(new_ticks)
plt.xticks(new_ticks)
plt.yticks([-2, -1.8, -1, 1.22, 3, ],
[r'$really\ bad$', r'$bad\ \alpha$', r'$normal$',
r'$good$', r'$really\ good$'])
l1, = plt.plot(x, y2)
l2, = plt.plot(x, y1, color="red", linewidth=1.0, linestyle="--")
plt.legend(handles=[l1, l2], labels=['aaa', 'bbb'], loc='best')
plt.show()
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/91f9ccd45d1466dd820902fa61eb2d90.png)