使用 GridSpec 自定义子图位置

原文：Customizing Location of Subplot Using GridSpec

译者：飞龙

协议：CC BY-NC-SA 4.0

GridSpec

指定子图将放置的网格的几何位置。 需要设置网格的行数和列数。 子图布局参数（例如，左，右等）可以选择性调整。

SubplotSpec

指定在给定GridSpec中的子图位置。

subplot2grid

一个辅助函数，类似于pyplot.subplot，但是使用基于 0 的索引，并可使子图跨越多个格子。

subplot2grid基本示例

要使用subplot2grid，你需要提供网格的几何形状和网格中子图的位置。 对于简单的单网格子图：

  1. ax = plt.subplot2grid((2,2),(0, 0))

等价于：

  1. ax = plt.subplot(2,2,1)
  1.        nRow=2, nCol=2
  2. (0,0) +-------+-------+
  3.       |   1   |       |
  4.       +-------+-------+
  5.       |       |       |
  6.       +-------+-------+

要注意不想subplot，gridspec中的下标从 0 开始。

为了创建跨越多个格子的子图，

  1. ax2 = plt.subplot2grid((3,3), (1, 0), colspan=2)
  2. ax3 = plt.subplot2grid((3,3), (1, 2), rowspan=2)

例如，下列命令：

  1. ax1 = plt.subplot2grid((3,3), (0,0), colspan=3)
  2. ax2 = plt.subplot2grid((3,3), (1,0), colspan=2)
  3. ax3 = plt.subplot2grid((3,3), (1, 2), rowspan=2)
  4. ax4 = plt.subplot2grid((3,3), (2, 0))
  5. ax5 = plt.subplot2grid((3,3), (2, 1))

会创建：

使用 GridSpec 自定义子图位置 - 图1

GridSpec和SubplotSpec

你可以显式创建GridSpec并用它们创建子图。

例如，

  1. ax = plt.subplot2grid((2,2),(0, 0))

等价于：

  1. import matplotlib.gridspec as gridspec
  2. gs = gridspec.GridSpec(2, 2)
  3. ax = plt.subplot(gs[0, 0])

gridspec示例提供类似数组（一维或二维）的索引，并返回SubplotSpec实例。例如，使用切片来返回跨越多个格子的SubplotSpec实例。

上面的例子会变成：

  1. gs = gridspec.GridSpec(3, 3)
  2. ax1 = plt.subplot(gs[0, :])
  3. ax2 = plt.subplot(gs[1,:-1])
  4. ax3 = plt.subplot(gs[1:, -1])
  5. ax4 = plt.subplot(gs[-1,0])
  6. ax5 = plt.subplot(gs[-1,-2])

使用 GridSpec 自定义子图位置 - 图2

调整 GridSpec布局

在显式使用GridSpec的时候，你可以调整子图的布局参数，子图由gridspec创建。

  1. gs1 = gridspec.GridSpec(3, 3)
  2. gs1.update(left=0.05, right=0.48, wspace=0.05)

这类似于subplots_adjust，但是他只影响从给定GridSpec创建的子图。

下面的代码

  1. gs1 = gridspec.GridSpec(3, 3)
  2. gs1.update(left=0.05, right=0.48, wspace=0.05)
  3. ax1 = plt.subplot(gs1[:-1, :])
  4. ax2 = plt.subplot(gs1[-1, :-1])
  5. ax3 = plt.subplot(gs1[-1, -1])
  7. gs2 = gridspec.GridSpec(3, 3)
  8. gs2.update(left=0.55, right=0.98, hspace=0.05)
  9. ax4 = plt.subplot(gs2[:, :-1])
  10. ax5 = plt.subplot(gs2[:-1, -1])
  11. ax6 = plt.subplot(gs2[-1, -1])

会产生

使用 GridSpec 自定义子图位置 - 图3

使用 SubplotSpec创建 GridSpec

你可以从SubplotSpec创建GridSpec，其中它的布局参数设置为给定SubplotSpec的位置的布局参数。

  1. gs0 = gridspec.GridSpec(1, 2)
  3. gs00 = gridspec.GridSpecFromSubplotSpec(3, 3, subplot_spec=gs0[0])
  4. gs01 = gridspec.GridSpecFromSubplotSpec(3, 3, subplot_spec=gs0[1])

使用 GridSpec 自定义子图位置 - 图4

使用SubplotSpec创建复杂嵌套的GridSpec

这里有一个更复杂的嵌套gridspec的示例，我们通过在每个 3x3 内部网格中隐藏适当的脊线，在 4x4 外部网格的每个单元格周围放置一个框。

使用 GridSpec 自定义子图位置 - 图5

网格尺寸可变的GridSpec

通常，GridSpec创建大小相等的网格。你可以调整行和列的相对高度和宽度，要注意绝对高度值是无意义的，有意义的只是它们的相对比值。

  1. gs = gridspec.GridSpec(2, 2,
  2.                        width_ratios=[1,2],
  3.                        height_ratios=[4,1]
  4.                        )
  6. ax1 = plt.subplot(gs[0])
  7. ax2 = plt.subplot(gs[1])
  8. ax3 = plt.subplot(gs[2])
  9. ax4 = plt.subplot(gs[3])

使用 GridSpec 自定义子图位置 - 图6