matplotlib的gridspec模块是专门指定画布中子图位置的模块,该模块中包含一个GridSpec类,通过显式地创建GridSpec类对象来自定义画布中子图的布局结构,使得子图能够更好地适应画布。
GridSpec常用参数的含义如下:
nrows:表示行数
ncols:表示列数
figure:表示布局的画布
left,bottom,right,top:表示子图的范围
wspace / hspace:表示子图之间预留的宽度量 / 高度量
width_ratios / height_ratios:设置坐标系的相对宽度 / 相对高度(每个坐标系的宽度 / 高度=设定值/sum)
下面就来介绍如何使用GridSpec()方法创建子图的布局结构:
这种方法需要创建子图和GridSpec类对象,之后在调用add_subplot()方法时传入GridSpec类对象即可。
创建一个两行两列的子图布局
首先:导入所需要用的库
import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec
第二步:创建画布实例
fig2 = plt.figure()
第三步:创建“区域规划图”实例(GridSpec实例)
spec2 = gridspec.GridSpec(ncols=2, nrows=2, figure=fig2)
第四步:根据给定的“区域规划图”,创建对应的坐标系实例
f2_ax1 = fig2.add_subplot(spec2[0,0]) # 二维数组的切片操作
f2_ax2 = fig2.add_subplot(spec2[0,1])
f2_ax3 = fig2.add_subplot(spec2[1,0])
f2_ax4 = fig2.add_subplot(spec2[1,1])
#展示图表
plt.show()
在第四步中可以看出,GridSpec类对象的使用方式和数组的使用方式相似,采用索引或者切片的形式访问每个布局元素。
上述运行结果如下:
这是一个两行两列的子图布局,主要控制参数就是nrows和ncols。我们还可以把所有的子图“打包”在一起,整体进行伸缩或者平移,调整子图的范围,控制参数就是left,bottom,right,top
为了方便查看子图位置变化,在导入库的上面添加一行%matplotlib notebook(在控制台嵌入GUI交互窗口,显示图表)
还有:
%matplotlib inline(控制台直接输出显示图表)
%matplotlib auto(弹出GUI交互窗口,显示图表)
通过窗口的形式显示图片,可以很好地体现子图与整个画布的位置关系
调整子图的范围
现在只需要在原来的代码上在第三步添加参数控制,修改后的第三步如下:
spec2 = gridspec.GridSpec(ncols=2, nrows=2, figure=fig2,
left=0.25,right=0.9,top=0.9,bottom=0.25) #把所有的子图“打包”在一起,整体进行伸缩或者平移
运行结果如下:
可以看出,子图整体是往布局的右上角移动了的。要注意:参数的数据表示子图范围的起始位置哦
设置坐标系的相对宽度和相对高度
有个值得注意的点,上面创建的子图都是等大的,那如何控制子图的比例大小呢?
可以通过width_ratios和height_ratios控制,同样是只修改第三步:
spec2 = gridspec.GridSpec(ncols=2, nrows=2, figure=fig2,
left=0.25,right=0.9,top=0.9,bottom=0.25, #把所有的子图“打包”在一起,整体进行伸缩或者平移
width_ratios=[3,1], height_ratios=[3,1]) #设置坐标系的相对宽度和相对高度
运行结果如下:
分成的四个子图中,宽高都分成了四等分。宽度上第一列子图占3/4,第二列子图占1/4,同理高度上第一行占三分,第二行占一分
设置坐标系实例之间纵向和横向间距
不知道大家是否留意到子图与子图间的距离过小,那想要控制子图间的间隔距离就可以通过wspace和hspace来实现,还是在第三步添加哦:
spec2 = gridspec.GridSpec(ncols=2, nrows=2, figure=fig2,
left=0.25,right=0.9,top=0.9,bottom=0.25, #把所有的子图“打包”在一起,整体进行伸缩或者平移
width_ratios=[3,1], height_ratios=[3,1], #设置坐标系的相对宽度和相对高度
wspace=2,hspace=1) #设置坐标系实例之间纵向和横向间距
运行结果如下:
以上就是GridSpec的具体使用,有疑问欢迎在评论区进行讨论^_^