Matplotlib GridSpec: 创建子图网格的指南

Matplotlib GridSpec: 创建子图网格的指南

'GridSpec' 是 Matplotlib 中用于创建均匀网格的类。它允许您在图中创建具有不同大小和位置的子图。

使用 GridSpec

1. 导入 GridSpec:

from matplotlib.gridspec import GridSpec

2. 创建 GridSpec 对象:

定义网格的行和列数。例如，创建一个具有 3 行和 2 列的网格：

gs = GridSpec(3, 2)

3. 访问子图位置:

使用索引访问每个子图的位置。例如，在网格的第一行第一列创建一个子图：

ax1 = plt.subplot(gs[0, 0])

您也可以使用索引访问其他子图的位置。例如，在网格的第二行第二列创建一个子图：

ax2 = plt.subplot(gs[1, 1])

4. 绘制子图:

使用 ax1、ax2 等对象设置和绘制每个子图。

示例：

import matplotlib.pyplot as plt
from matplotlib.gridspec import GridSpec

# 创建一个 2x2 的 GridSpec
gs = GridSpec(2, 2)

# 创建四个子图
ax1 = plt.subplot(gs[0, 0])
ax2 = plt.subplot(gs[0, 1])
ax3 = plt.subplot(gs[1, 0])
ax4 = plt.subplot(gs[1, 1])

# 在子图中绘制内容
ax1.plot([1, 2, 3], [4, 5, 6])
ax2.scatter([1, 2, 3], [4, 5, 6])
ax3.bar([1, 2, 3], [4, 5, 6])
ax4.hist([1, 2, 3, 4, 5, 6])

# 显示图形
plt.show()

通过 GridSpec，您可以灵活地创建各种子图布局，满足您的可视化需求。