为你的图片添加图例

欢迎关注”生信修炼手册”!

图例可以帮助我们更好的理解图中的信息，在matplotlib中，通过legend函数来添加图例，有以下两种用法

1. 在绘制元素时指定label,然后legend自动识别对应的label属性，绘制图例

2. 绘制元素时不需要指定label, 但是需要获取对应的artist对象，然后在legend函数中为其指定对应的label

具体实现的代码如下

# 第一种方法
>>> x = np.linspace(0, 2 * np.pi, 50)
>>> plt.plot(x, np.sin(x), label='sin')
>>> plt.plot(x, np.cos(x), label='cos')
>>> plt.legend()
>>> plt.show()
`
# 第二种方法
>>> x = np.linspace(0, 2 * np.pi, 50)
>>> line1, = plt.plot(x, np.sin(x))
>>> line2, = plt.plot(x, np.cos(x))
>>> plt.legend((line1, line2), ('sin', 'cos'))
>>> plt.show()

两种方法都可以实现如下所示的效果

当图例中只需要展示一个元素时，还有一种特殊用法，代码如下

>>> x = np.linspace(0, 2 * np.pi, 50)
>>> plt.plot(x, np.sin(x))
>>> plt.legend(['sin'])
>>> plt.show()

输出效果如下

对于图例而言，我们最常修改的属性就是图例的位置了，在matplotlib中，可以通过以下loc参数来调整图例的位置，有两种设置方式，第一种用表示位置的字符串来定义图例位置，纵向的位置用lower,center, upper表示，横向的位置用left ,center,right表示，二者组合起来共同定义图例的位置，用法如下

>>> x = np.linspace(0, 2 * np.pi, 50)
>>> plt.plot(x, np.sin(x))
>>> plt.legend(['sin'], loc='upper center')
>>> plt.show()

输出结果如下

采用这种赋值方式时，有两种特殊取值，第一个为center,表示无论在水平还是垂直方向上，都位于图片的正中心，第二个是best, 也是默认取值，以不和已有元素重叠为标准，自动计算最佳的图例位置。

第二种设置方式通过两个0到1的浮点数，来指定图例左下