前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >为你的图片添加图例

为你的图片添加图例

作者头像
生信修炼手册
发布2020-09-14 11:02:04
1.2K0
发布2020-09-14 11:02:04
举报
文章被收录于专栏:生信修炼手册生信修炼手册

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

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

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

具体实现的代码如下

代码语言:javascript
复制
# 第一种方法
>>> 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()

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

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

代码语言:javascript
复制
>>> 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表示,二者组合起来共同定义图例的位置,用法如下

代码语言:javascript
复制
>>> 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的浮点数,来指定图例左下角在axes中的位置,将axes的左下角视为(0, 0), 将右上角视为(1, 1), 用法如下

代码语言:javascript
复制
>>> x = np.linspace(0, 2 * np.pi, 50)
>>> plt.plot(x, np.sin(x))
>>> plt.legend(['sin'], loc=(0.65, 0.75))
>>> plt.show()

输出结果如下

legend函数实际上有两种方法,axes,legend和figure.legend, 上面的代码都是调用的axes.legend, 所以实在axes的范围内设置图例,当我们想要实现图例在axes之外时,可以通过figure.legend方法来实现,用法如下

代码语言:javascript
复制
>>> fig = plt.figure()
>>> ax = plt.axes()
>>> x = np.linspace(0, 2 * np.pi, 50)
>>> ax.plot(x, np.sin(x))
>>> fig.legend(['sin'], loc=(0.55, 0.9))
>>> plt.show()

输出结果如下

通过对legend的返回值进行操作,也可以个性化的定义图例的样式,常见的用法如下

代码语言:javascript
复制
>>> x = np.linspace(0, 2 * np.pi, 50)
>>> plt.plot(x, np.sin(x), label='sin')
>>> plt.plot(x, np.cos(x), label='cos')
>>> legend = plt.legend()
>>> legend
<matplotlib.legend.Legend object at 0x0943ACB8>
>>> legend.texts
[Text(0, 0, 'sin'), Text(0, 0, 'cos')]
>>> legend.texts[0].set_color('r')
>>> legend.texts[1].set_color('b')
>>> legend.get_frame().set_facecolor('c')
>>> plt.show()

用texts捕获图例中的文字,用get_frame方法返回图例的边框区域,上述代码的输出结果如下

legend的参数还有很多,详细用法请参考官方文档。

·end·

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2020-09-07,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 生信修炼手册 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档