如何在matplotlib中加注释和内嵌图

开发 后端
这篇文章会以一个实际的案例,详细讲讲如何给数据加注解,同时也介绍一下一种比较骚的操作,即Matplotlib的内嵌图(把一张小图嵌入到一张大图中),学会这个后,你能够绘制出这种图形。

[[331954]]

 在用Matplotlib进行可视化过程中,很多时候为了更直观地展现数据大小,会将具体的数值标注在图形中,比如在柱状图上标明数值大小。

这篇文章会以一个实际的案例,详细讲讲如何给数据加注解,同时也介绍一下一种比较骚的操作,即Matplotlib的内嵌图(把一张小图嵌入到一张大图中),学会这个后,你能够绘制出这种图形

数据注释

演示的数据集为2016年抵美(到达美国)人数排名前十的国家数据,包含国名和具体入境人数,人数的具体单位为百万人次 

  1. import pandas as pd  
  2. import numpy as np  
  3. import matplotlib.pyplot as plt  
  4. %matplotlib inline  
  5. plt.rcParams['font.sans-serif'] = 'SimHei' 
  6. x_data = ['加拿大','墨西哥','英国','日本','中国','德国','韩国','法国','巴西','印度']  
  7. y_data = [13.428347,12.255948,3.019111,2.352919,2.09833,1.322882,1.282097,1.127868,1.109066,0.874016] 

有了数据后,可以非常快地画出一张柱状图 

  1. fig,ax = plt.subplots(figsize=(16,6))  
  2. ax.bar(x_data,y_data) 

接下来开始加注释,即在柱状图上显示具体数值。在Matplotlib中,为数据加上注释有两种方式,一种是使用ax.text(),另一种则是ax.annotate()。

ax.text()

ax.text()的主要作用是为图中加上一些text,也就是文字。不仅是能够加注释,只要指定了坐标,可以在图上的任何坐标加上text。

函数中的几个重要的参数,具体介绍一下

  •  x:x的坐标
  •  y:y的坐标
  •  s:要加的文字
  •  rotation:文字旋转的角度
  •  fontsize:文字字体大小
  •  fontweight:文字字体粗细

需要注意的是,每次调用ax.text()只能生成一个Text对象,也就是说每次只能加一个注释,多个的话需要写循环生成。

所以一般加注释是这么一个流程:先确定注释的横纵坐标-->写循环调用ax.text() 

  1. fig,ax = plt.subplots(figsize=(16,6))  
  2. ax.bar(x_data,y_data)  
  3. # 循环生成text  
  4. # horizontalalignment参数设置注释居中显示  
  5. for x,y in zip(x_data,y_data):  
  6.     ax.text(x,y+0.05,y,fontsize=14,horizontalalignment='center'

关于图表美化方面,这里暂时不过多赘述。

ax.annotate()

相比于ax.text(),ax.annotate()更像是专门为做注释而生的,annotate便是注释的意思。

ax.annotate()的注释功能更强大,除了加入文本注释外,如果有需要还能够加上箭头→进行指示。

一些主要的参数如下:

  •  s:注释文本
  •  xy:要加注释的数据点位置
  •  xytext:文本注释的位置,默认情况下为xy
  •  arrowprops:一个控制箭头的属性的dict,如果需要显示箭头,必须要设置

这里比较有意思的两个参数是xy和xytext,二者貌似没啥区别的样子,这个地方确实很容易产生困惑。

一般情况下,s和xy是必须要设置的参数,如果不指定,xytext默认和xy一致。

但如果要设置箭头的话,xy的坐标则定义了箭头的头部,xytext则指定箭头的尾部和文本注释的位置,实际画图来理解看看。

ax.annotate()和ax.text()的画图流程是一致的,都需要循环生成注释。

不设置箭头,简单加上注释 

  1. fig,ax = plt.subplots(figsize=(16,6))  
  2. ax.bar(x_data,y_data) 
  3. for x,y in zip(x_data,y_data):  
  4.     ax.annotate(y,(x,y+0.05),fontsize=14,horizontalalignment='center'

加上注释并设置箭头 

  1. fig,ax = plt.subplots(figsize=(16,6))  
  2. ax.bar(x_data,y_data)  
  3. # 箭头头部坐标(x,y)  
  4. # 箭头尾部坐标(注释坐标)为(x,y+1)  
  5. for x,y in zip(x_data,y_data):  
  6.     ax.annotate(y,xy = (x,y),xytext = (x,y+1),fontsize=14,horizontalalignment='center' 
  7.                arrowprops =dict(arrowstyle='->'))  
  8. # 将纵坐标范围扩大  
  9. ax.set_ylim([0,16]) 

设置了显示箭头之后,可以明显地看出,箭头是由xytext坐标指向xy的坐标的,所以,当你不需要设置箭头的时候,xytext设置的意义并不大。

内嵌图

内嵌图大家可能用的比较少,但这种图其实还是挺有用的。

拿上面的数据例子来说,柱状图显示了2016年抵美人数排名前十的国家的具体人数,如果这时想结合各大地域抵美人数的占比数据进行分析,该如何绘图?

一个比较直接的想法是用subplots子图来实现,比如上边显示柱状图,下边显示饼图,如下: 

  1. zhou_name = ['西欧','亚洲','南美洲','大洋洲','加勒比地区','中东地区','中美洲','东欧','非洲']  
  2. zhou_percent = [36.2,30.8,13.9,4.3,4.1,3.8,2.8,2.6,1.5]  
  3. fig,ax = plt.subplots(2,1,figsize=(16,12))  
  4. ax[0].bar(x_data,y_data)  
  5. for x,y in zip(x_data,y_data):  
  6.     ax[0].annotate(y,(x,y+0.05),fontsize=14,horizontalalignment='center' 
  7. ax[1].pie(zhou_percent,labels=zhou_name,autopct='%1.1f%%'

这当然是可行的,但还有更好的方案,那便是内嵌图。注意到这里的柱状图右侧有很大的留白部分,如果把饼图放到柱状图右侧的留白部分会显得更加直观。

内嵌图有两种生成方式,一种是fig.add_axes(),另一种则是使用inset_axes()。

fig.add_axes()

fig.add_axes()就是在原有的Figure上加上一个新的区域Axes,然后在这个区域中绘制图形。

使用这个方法的话需要指定新增的这个区域Axes在Figure中的相对位置和区域大小,输入参数均为相对于原来Figure的比例值,如下: 

  1. # left和bottom控制新Axes的位置  
  2. # width和height控制新Axes的大小(长宽)  
  3. # 这些均用相对数来表示,大小在0-1之间  
  4. left,bottom,width,height = [0.5,0.3,0.5,0.5]  
  5. fig,ax1 = plt.subplots(figsize=(16,6))  
  6. ax1.bar(x_data,y_data)  
  7. for x,y in zip(x_data,y_data):  
  8.     ax1.annotate(y,(x,y+0.05),fontsize=14,horizontalalignment='center' 
  9. # 运用fig.add_axes()新增一个区域Axes绘图  
  10. ax2 = fig.add_axes([left,bottom,width,height])  
  11. ax2.pie(zhou_percent,labels=zhou_name,autopct='%1.1f%%'

inset_axes

相比于fig.add_axes()需要对相对位置进行调试,使用inset_axes()进行绘图则可以方便进行定位。

inset_axes中的位置由参数loc设置,可以用字符串控制,也可以输入数字,具体如下: 

  1. 'upper right'  : 1  
  2. 'upper left'   : 2  
  3. 'lower left'   : 3 
  4. ...... 
  5. 'upper center' : 9  
  6. 'center'       : 10  
  1. # 使用前需要先导包  
  2. from mpl_toolkits.axes_grid1.inset_locator import inset_axes  
  3. fig,ax1 = plt.subplots(figsize=(16,6))  
  4. ax1.bar(x_data,y_data)  
  5. for x,y in zip(x_data,y_data):  
  6.     ax1.annotate(y,(x,y+0.05),fontsize=14,horizontalalignment='center' 
  7. # 将内嵌图置于右侧,宽度和长度分别为相对长度  
  8. ax2 = inset_axes(ax1,width = '60%',height = '60%',loc='right' 
  9. ax2.pie(zhou_percent,labels=zhou_name,autopct='%1.1f%%'

美化方面这里就不再多讲了,具体可以参照这篇文章如何用 Matplotlib 画一张好看的图。 

 

责任编辑:庞桂玉 来源: Python中文社区
相关推荐

2023-10-19 14:52:27

2023-11-30 20:51:26

多子图布局matplotlib

2017-02-08 21:20:24

LinuxUbuntu加密硬盘

2013-11-13 09:22:16

2013-04-02 15:23:12

2024-03-25 00:02:00

Android移动开发视频

2023-08-28 15:52:10

LinuxPDF

2023-01-18 17:22:38

YAML注释Linux

2020-02-24 13:06:55

Python数据帧开发

2022-09-26 08:35:53

磁盘Java解密

2013-04-01 15:15:35

Eclipse自动添加Android开发

2009-11-24 19:33:07

PHP字符串中加入变量

2012-11-30 16:02:57

在vSphereiSCSI存储虚拟化

2011-08-01 15:20:51

SQL Server索引

2011-10-13 16:34:30

MeeGoQt插件

2011-03-02 17:41:28

2011-03-18 09:48:32

2010-09-07 14:22:41

SQL语句

2015-08-05 09:40:26

2020-05-19 13:55:38

Python加密密码
点赞
收藏

51CTO技术栈公众号