在一个Figure中安排多个Axes
通常在一个图像中,需要同时呈现多于一个Axes,并且需要对齐到网格. Matplotlib有多种工具用于处理在本库历史中演变的Axes网格,我们将讨论我们认为用户最常使用的工具,支持Axes组织方式的工具,并提及一些较旧的工具.
注意: Matplotlib使用Axes来引用包含数据、x轴和y轴、刻度、标签、标题等的绘图区域. 有关详细信息,请参阅Parts of a Figure这一节. 另一个经常使用的术语是“subplot”,它指的是与其他Axes对象一起位于Axes风格中的一个Axes.
0 总览
0.1 创建轴的网格形状组合
subplots
用于创建图形和Axes网格的主要功能。它一次创建所有Axes并将其放置在图形上,并返回一个对象数组,其中包含风格中Axes的句柄,参见Figure.subplots
.
或者
subplot_mosaic
一种创建图形和Axes风格的简单方法,具有额外的灵活性,即Axes也可以跨越行和列。Axes在带标签的字典中返回,而不是在数组中返回. 另外请参阅Figure.subplot_mosaic
以及复杂和语义图形组合(subplot_mosaic).
有时,拥有多组不同的Axes网格是很自然的,在这种情况下,Matplotlib具有SubFigure的概念:
SubFigure
一个虚拟的在一个Figure内的Figure.
0.2 基础工具
这些是GridSpec
和SubplotSpec
的概念:
GridSpec
指定将旋转子图的网格的几何图形. 需要设置网格的行数和列数. 或者,可以调整子图布局参数(例如,左、右等).
SubplotSpec
在给定GridSpec
中指定子图的位置.
0.3 一次添加一个Axes
上述函数在单个函数调用中创建所有的Axes,也可以一次添加一个Axes,这最初是Matplotlib的工作方式,这样做通常不太优雅和灵活,但有时对于交互式工作或将Axes旋转在自定义位置很有用.
add_axes
在[left, bottom, width, height]指定的位置添加单个Axes,以图形宽度或高度的小数表示.
subplot
或者Figure.add_subplot
在图形上添加一个子图,使用自1开始的索引(继承自Matlab的方式),可以通过指定网格单元格的范围来跨越行和列.
subplot2grid
与pyplot.subplot
类似,但使用自0开始的索引和二维python切片来选择单元格. 作为手动添加Axes a的简单示例,让我们将3英寸×2英寸的Axes添加到4英寸×3英寸的图形中. 请注意,子图的位置以图形归一化单位定义为[left, bottom, width, height]:
import matplotlib.pyplot as plt
import numpy as npw, h = 4, 3
margin = 0.5
fig = plt.figure(figsize=(w, h), facecolor='lightblue')
ax = fig.add_axes([margin / w, margin / h, (w - 2 * margin) / w,(h - 2 * margin) / h])
1 生成网格的高级方法
1.1 基础2×2网格
我们可以使用子图创建一个基本的2×2Axes网格.它返回一个Figure实例和一个Axes对象数组. Axes对象可用于访问将artists放置在Axes上的方法;这里我们使用annotate,但其他示例可能是plot、pcolormesh等.
fig, axs = plt.subplots(ncols=2, nrows=2, figsize=(5.5, 3.5),layout="constrained")
# add an artist, in this case a nice label in the middle...
for row in range(2):for col in range(2):axs[row, col].annotate(f'axs[{row}, {col}]', (0.5, 0.5),transform=axs[row, col].transAxes,ha='center', va='center', fontsize=18,color='darkgrey')
fig.suptitle('plt.subplots()')
我们将对很多Axes进行注释,因此让我们封装annotate, 而不是每次需要时都使用那一大段注释代码:
def annotate_axes(ax, text, fontsize=18):ax.text(0.5, 0.5, text, transform=ax.transAxes,ha="center", va="center", fontsize=fontsize, color="darkgrey")
使用subplot_mosaic
可以实现相同的效果, 但返回类型是字典而不是数组, 用户可以在其中为键提供有用的含义. 这里我们提供了两个列表, 每个列表代表一行, 列表中的每个元素代表列的键.
fig, axd = plt.subplot_mosaic([['upper left', 'upper right'],['lower left', 'lower right']],figsize=(5.5, 3.5), layout="constrained")
for k, ax in axd.items():annotate_axes(ax, f'axd[{k!r}]', fontsize=14)
fig.suptitle('plt.subplot_mosaic()')
1.2 固定纵横比Axes的网格
固定纵横比Axes对于图像或地图很常见. 但是, 它们对布局提出了挑战, 因为对Axes的大小施加了两组约束——它们适合图形, 并且它们具有设定的纵横比. 默认情况下, 这会导致Axes之间的间隙很大:
fig, axs = plt.subplots(2, 2, layout="constrained",figsize=(5.5, 3.5), facecolor='lightblue')
for ax in axs.flat:ax.set_aspect(1)
fig.suptitle('Fixed aspect Axes')
解决此问题的一种方法是将图形的纵横比更改为接近Axes的纵横比, 但这需要反复试验. Matplotlib还提供了 layout=“compressed”, 它将与简单的网格一起使用, 以减少Axes之间的间隙(mpl_toolkits还提供了ImageGrid来实现类似的效果,但使用非标准的Axes类).
fig, axs = plt.subplots(2, 2, layout="compressed", figsize=(5.5, 3.5),facecolor='lightblue')
for ax in axs.flat:ax.set_aspect(1)
fig.suptitle('Fixed aspect Axes: compressed')
1.3 Axes跨越网格中的行或列
有时我们希望Axes跨越网格的行或列. 实际上有多种方法可以实现此目的, 但最方便的方法可能是通过重复其中一个键来使用subplot_mosaic:
fig, axd = plt.subplot_mosaic([['upper left', 'right'],['lower left', 'right']],figsize=(5.5, 3.5), layout="constrained")
for k, ax in axd.items():annotate_axes(ax, f'axd[{k!r}]', fontsize=14)
fig.suptitle('plt.subplot_mosaic()')
有关如何使用GridSpec或subplot2grid执行相同操作的说明, 请参阅下文.
1.4 网格中的可变宽度或高度
子图和subplot_mosaic都允许网格中的行具有不同的高度,并且使用gridspec_kw关键字参数将列设置为不同的宽度. GridSpec接受的间距参数可以传递给子图, 并 subplot_mosaic:
gs_kw = dict(width_ratios=[1.4, 1], height_ratios=[1, 2])
fig, axd = plt.subplot_mosaic([['upper left', 'right'],['lower left', 'right']],gridspec_kw=gs_kw, figsize=(5.5, 3.5),layout="constrained")
for k, ax in axd.items():annotate_axes(ax, f'axd[{k!r}]', fontsize=14)
fig.suptitle('plt.subplot_mosaic()')
1.5 嵌套Axes布局
有时, 拥有两个或多个可能不需要相互关联的Axes网格会很有帮助. 实现此目的的最简单方法是使用Figure.subfigures. 请注意,子图布局是独立的, 因此每个子图中的Axes中心不一定对齐. 请参阅下文, 了解使用 GridSpecFromSubplotSpec实现相同效果的更详细方法.
fig = plt.figure(layout="constrained")
subfigs = fig.subfigures(1, 2, wspace=0.07, width_ratios=[1.5, 1.])
axs0 = subfigs[0].subplots(2, 2)
subfigs[0].set_facecolor('lightblue')
subfigs[0].suptitle('subfigs[0]\nLeft side')
subfigs[0].supxlabel('xlabel for subfigs[0]')axs1 = subfigs[1].subplots(3, 1)
subfigs[1].suptitle('subfigs[1]')
subfigs[1].supylabel('ylabel for subfigs[1]')
也可以使用嵌套列表subplot_mosaic嵌套Axes. 此方法不使用子图, 如上所述, 因此无法添加每个子图的suptitle 和 supxlabel等. 相反, 它是下面描述的 subgridspec 方法的方便包装器.
inner = [['innerA'],['innerB']]
outer = [['upper left', inner],['lower left', 'lower right']]fig, axd = plt.subplot_mosaic(outer, layout="constrained")
for k, ax in axd.items():annotate_axes(ax, f'axd[{k!r}]')
2. 低级和高级的网格方法
在内部, Axes网格的排列是通过创建GridSpec和SubplotSpec的实例来控制的. GridSpec定义了一个(可能是非均匀的)单元格网格. 索引到GridSpec中将返回一个SubplotSpec, 该SubplotSpec涵盖一个或多个网格单元格, 可用于指定Axis的位置.
下面的示例演示如何使用低级方法通过GridSpec对象排列坐标区.
2.1 基础2×2网格
我们可以用与plt.subplots(2,2)相同的方式完成一个2×2网格:
fig = plt.figure(figsize=(5.5, 3.5), layout="constrained")
spec = fig.add_gridspec(ncols=2, nrows=2)ax0 = fig.add_subplot(spec[0, 0])
annotate_axes(ax0, 'ax0')ax1 = fig.add_subplot(spec[0, 1])
annotate_axes(ax1, 'ax1')ax2 = fig.add_subplot(spec[1, 0])
annotate_axes(ax2, 'ax2')ax3 = fig.add_subplot(spec[1, 1])
annotate_axes(ax3, 'ax3')fig.suptitle('Manually added subplots using add_gridspec')
2.2 Axes跨越网格中的行或网格
我们可以使用 Numpy slice语法索引spec数组, 新的Axes将跨越该切片. 这与fig, axd = plt.subplot_mosaic([['ax0', 'ax0'], ['ax1', 'ax2']], ...)
效果一样:
fig = plt.figure(figsize=(5.5, 3.5), layout="constrained")
spec = fig.add_gridspec(2, 2)ax0 = fig.add_subplot(spec[0, :])
annotate_axes(ax0, 'ax0')ax10 = fig.add_subplot(spec[1, 0])
annotate_axes(ax10, 'ax10')ax11 = fig.add_subplot(spec[1, 1])
annotate_axes(ax11, 'ax11')fig.suptitle('Manually added subplots, spanning a column')
2.3 手动调整GridSpec布局
显式使用GridSpec时, 可以调整从GridSpec创建的子图的布局参数. 请注意, 此选项与受约束的布局或 Figure.tight_layout不兼容, 后者会忽略左侧和右侧, 并调整子图的大小以填充图. 通常, 这种手动放置需要迭代, 以使Axes刻度标签不与Axes重叠.
这些间距参数也可以传递给子图, 并作为gridspec_kw参数subplot_mosaic.
fig = plt.figure(layout=None, facecolor='lightblue')
gs = fig.add_gridspec(nrows=3, ncols=3, left=0.05, right=0.75,hspace=0.1, wspace=0.05)
ax0 = fig.add_subplot(gs[:-1, :])
annotate_axes(ax0, 'ax0')
ax1 = fig.add_subplot(gs[-1, :-1])
annotate_axes(ax1, 'ax1')
ax2 = fig.add_subplot(gs[-1, -1])
annotate_axes(ax2, 'ax2')
fig.suptitle('Manual gridspec with right=0.75')
2.4 使用SubplotSpec的嵌套布局
你可以使用subgridspec创建类似于子图形的嵌套布局. 在这里, Axes的中心是对齐的.
请注意, 这也可以从更详细的说明获得: gridspec.GridSpecFromSubplotSpec
fig = plt.figure(layout="constrained")
gs0 = fig.add_gridspec(1, 2)gs00 = gs0[0].subgridspec(2, 2)
gs01 = gs0[1].subgridspec(3, 1)for a in range(2):for b in range(2):ax = fig.add_subplot(gs00[a, b])annotate_axes(ax, f'axLeft[{a}, {b}]', fontsize=10)if a == 1 and b == 1:ax.set_xlabel('xlabel')
for a in range(3):ax = fig.add_subplot(gs01[a])annotate_axes(ax, f'axRight[{a}, {b}]')if a == 2:ax.set_ylabel('ylabel')fig.suptitle('nested gridspecs')
下面是一个更复杂的嵌套GridSpec示例: 我们创建一个外部4×4网格, 每个单元格都包含一个内部3×3 Axes 网格. 我们通过在每个内部3×3网格中隐藏适当的脊柱来勾勒外部4×4网格.
def squiggle_xy(a, b, c, d, i=np.arange(0.0, 2*np.pi, 0.05)):return np.sin(i*a)*np.cos(i*b), np.sin(i*c)*np.cos(i*d)fig = plt.figure(figsize=(8, 8), layout='constrained')
outer_grid = fig.add_gridspec(4, 4, wspace=0, hspace=0)for a in range(4):for b in range(4):# gridspec inside gridspecinner_grid = outer_grid[a, b].subgridspec(3, 3, wspace=0, hspace=0)axs = inner_grid.subplots() # Create all subplots for the inner grid.for (c, d), ax in np.ndenumerate(axs):ax.plot(*squiggle_xy(a + 1, b + 1, c + 1, d + 1))ax.set(xticks=[], yticks=[])# show only the outside spines
for ax in fig.get_axes():ss = ax.get_subplotspec()ax.spines.top.set_visible(ss.is_first_row())ax.spines.bottom.set_visible(ss.is_last_row())ax.spines.left.set_visible(ss.is_first_col())ax.spines.right.set_visible(ss.is_last_col())plt.show()
3 更多内容
- 更多细节有关于 subplot mosaic.
- 更多关于constrained layout的细节, 在大多数示例中用于对齐间距.
参考
以下的函数、方法、类和模块显示在以下例子里:
- matplotlib.pyplot.subplots
- matplotlib.pyplot.subplot_mosaic
- matplotlib.figure.Figure.add_gridspec
- matplotlib.figure.Figure.add_subplot
- matplotlib.gridspec.GridSpec
- matplotlib.gridspec.SubplotSpec.subgridspec
- matplotlib.gridspec.GridSpecFromSubplotSpec