图例是啥,直接上图就知道了:
怎么创建上面的图例呢?
很简单,首先在plt.plot()函数中设置label文本属性,然后调用plt.legend()生成图例就可以了,完整的代码如下:
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
# 生成x轴上的数据:从-3到3,总共有50个点
x = np.linspace(-1, 1, 50)
# 定义一个线性方程
y1 = 2 * x + 1
# 定义一个二次方程
y2 = x ** 2
plt.plot(x, y2, label="y2")
# 绘制红色的线宽为1虚线的线条
plt.plot(x, y1, color=\'red\', linewidth=1.0, linestyle=\'--\', label="y1")
plt.legend()
# 显示图表
plt.show()
上面是最简单的生成图例的方法,在plt.legend()函数中还有其它的一些属性,我们以一个例子来显示,只要读者能够有些了解就可以了:
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
# 生成x轴上的数据:从-3到3,总共有50个点
x = np.linspace(-1, 1, 50)
# 定义一个线性方程
y1 = 2 * x + 1
# 定义一个二次方程
y2 = x ** 2
# 绘制红色的线宽为1虚线的线条,注意这里p1后面有个逗号
p1, = plt.plot(x, y1, color=\'red\', linewidth=1.0, linestyle=\'--\', label="y1")
p2, = plt.plot(x, y2, label="y2")
# handles用于指定显示哪些图例,labels可以再次重新指定显示图例的标签内容,loc定义图例显示的位置,best会自动寻找不挡住曲线的最佳位置,其余还有\'upper right\'、\'right\'等。
plt.legend(handles=[p1, p2], labels=["aaa", "bbb"], loc=\'best\')
# 显示图表
plt.show()
显示图为: