Matplotlib 中如何创建不同大小的子图 subplot
Jinku Hu
2024年2月15日
我们可以使用 gridspec_kw，gridspec 和 subplot2grid 指定不同比例的子图来创建不同大小的子图。
gridspec 方法
import matplotlib.pyplot as plt
from matplotlib import gridspec
fig = plt.figure()
spec = gridspec.GridSpec(ncols=2, nrows=1, width_ratios=[2, 1])
ax0 = fig.add_subplot(spec[0])
ax0.plot(range(5), range(5, 10))
ax1 = fig.add_subplot(spec[1])
ax1.plot(range(5), range(5, 10))
plt.show()
来自 gridspec 模块的 GridSpec 指定了子图网格的几何形状。我们可以设置行数，列数和布局参数，例如宽度和高度比。
spec = gridspec.GridSpec(ncols=2, nrows=1, width_ratios=[2, 1])
它指定图形具有两列和一行，并且宽度比为 2:1。
ax0 = fig.add_subplot(spec[0])
如上所述，ax0 对象在地物中的位置为 0，或者可以使用 spec[0, 0] 使它更清晰。
gridspec_kw 方法
import matplotlib.pyplot as plt
fig, ax = plt.subplots(
2, 2, gridspec_kw={"width_ratios": [2, 1], "height_ratios": [1, 2]}
)
ax[0][0].plot(range(5), range(5, 10))
ax[0][1].plot(range(5), range(10, 5, -1))
ax[1][0].plot(range(5), range(5, 10))
ax[1][1].plot(range(5), range(10, 5, -1))
plt.show()
gridspec_kw 是具有 GridSpec 构造函数关键字的字典，用于指定子图的网格。
宽度比指定为 2:1，高度比设置为 1:2。
subplot2grid 方法
subplot2grid 允许子图在网格索引中占用多个单元格。
def subplot2grid(shape,
loc,
rowspan=1,
colspan=1,
fig=None,
**kwargs):
shape 是网格的形状，具有 2 个整数的序列，其第一个元素是行数，第二个元素是列数。
loc 是 axis 放置在网格中的位置。它也是 2 个整数的序列，其第一个元素是行号，第二个元素是列号。
rowspan 和 colspan 是 axis 向右（rowspan）或向底部（colspan）跨越的行数或列数。
工作代码示例：
import matplotlib.pyplot as plt
fig = plt.figure()
ax0 = plt.subplot2grid((1, 5), (0, 0), colspan=3)
ax1 = plt.subplot2grid((1, 5), (0, 3), colspan=2)
ax0.plot(range(5), range(5, 10))
ax1.plot(range(5), range(10, 5, -1))
plt.show()
ax0 = plt.subplot2grid((1, 5), (0, 0), colspan=3)
ax1 = plt.subplot2grid((1, 5), (0, 3), colspan=2)
网格的形状为 (1, 5)，具有一行五列。
ax0 位于 (0, 0) 单元格中，占据 3 列，而 ax1 位于 (0, 3) 中，并且占据其他两列。
作者： Jinku Hu
