教程 - 使用 GridSpec 自定义子图位置

优质
小牛编辑
134浏览
2023-12-01

原文:Customizing Location of Subplot Using GridSpec

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. +-------+-------+

要注意不想subplotgridspec中的下标从 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

GridSpecSubplotSpec

你可以显式创建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])
  6. gs2 = gridspec.GridSpec(3, 3)
  7. gs2.update(left=0.55, right=0.98, hspace=0.05)
  8. ax4 = plt.subplot(gs2[:, :-1])
  9. ax5 = plt.subplot(gs2[:-1, -1])
  10. ax6 = plt.subplot(gs2[-1, -1])

会产生

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

使用 SubplotSpec创建 GridSpec

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

  1. gs0 = gridspec.GridSpec(1, 2)
  2. gs00 = gridspec.GridSpecFromSubplotSpec(3, 3, subplot_spec=gs0[0])
  3. 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. )
  5. ax1 = plt.subplot(gs[0])
  6. ax2 = plt.subplot(gs[1])
  7. ax3 = plt.subplot(gs[2])
  8. ax4 = plt.subplot(gs[3])

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