Chapter4 图例

Chapter4 图例

    图例的作用就是对所绘制的图像，进行解释。

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(-3,3,50)
y1 = 2*x+1
y2 = x**2

plt.figure()

# 设置x轴的取值范围
plt.xlim((-1,2))
# 设置y轴的取值范围
plt.ylim((-2,3))
# 描述x轴
plt.xlabel('I am x')
# 描述y轴
plt.ylabel('I am y')
# 设置新的角标
new_ticks = np.linspace(-1,2,5)
print(new_ticks)
plt.xticks(new_ticks)  # 把新的角标应用到x轴
# 设置y的角标，使得数字与相应注释一一对应
# 通过给注释首尾添加$符号，使得字体更加好看。当然，空格需要转义才可以显示
plt.yticks([-2,-1.8,-1,1.22,3],['$really\ bad$','$bad$','$normal$','$good$','$really\ good$'])

# label参数代表这个图像的标签，在设置图例时，如果labels参数为空，则自动选择此标签。
# plot的返回值是一个只有一个元素的列表，所以当我们想取其返回值时，需要解包。
l1, = plt.plot(x,y2,label='up')
l2, = plt.plot(x,y1,color='red',linewidth=1.0,linestyle='--',label='down')

# 设置图例
# loc 代表图例的位置，如果为best，则自动选择一个较好的位置
# handles 的参数是一个列表，里面的参数是plot方法所返回的对象，代表我这个图例要对其哪些图像作解释
# labels 的参数也是一个列表，代表对handles里面的图像做标签（解释）
plt.legend(handles=[l1,l2],labels=['1','2'],loc='best')

plt.show()