【莫烦Python】Matplotlib Python 画图教程 Legend图例

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(-1, 1, 50)  # 从负一到一生成五十个点
y1 = 2 * x + 1
y2 = x ** 2

plt.figure()
plt.plot(x, y1)  # 让x和y展示出来

plt.figure(num=2, figsize=(8, 5))  # num给图片定义名称，figsize给图片定义长宽
plt.plot(x, y2)  # 让x和y展示出来
plt.plot(x, y1, color='red', linewidth=1.0, linestyle='--')  # 在一张图片里显示另外一段线，红色，宽度1，虚线

plt.xlim((-1, 2))  # x轴取值范围
plt.ylim((-2, 3))  # y轴取值范围
plt.xlabel('I am x')
plt.ylabel('I am y')  # y轴取名

new_ticks = np.linspace(-1, 2, 5)  # 负一到二总共五个刻度
print(new_ticks)
plt.xticks(new_ticks)  # 换x轴的刻度

plt.yticks([-2, -1.8, -1, 1.22, 3, ],
           ['really bad', 'bad', 'normal', 'good', 'really good'])  # 将刻度修改成名称表示
l1,=plt.plot(x,y2,label='up')#给虚线或实线命名
l2,=plt.plot(x,y1,color='red',linewidth=1.0,linestyle='--',label='down')
#如果想用到legend里面的handles可以用l1,的形式来自定义标签
plt.legend(handles=[l1,l2,],labels=['aaa','bbb'],loc='best')#用legend显示出来

plt.show()  # 只有用到show才能让图显示出来