Matplotlib 中如何创建不同大小的子图 subplot

我们可以使用 gridspec_kwgridspecsubplot2grid 指定不同比例的子图来创建不同大小的子图。

Matplotlib 中如何创建不同大小的子图 subplot

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()

Matplotlib 中如何创建不同大小的子图 subplot

gridspec_kw 是具有 GridSpec 构造函数关键字的字典,用于指定子图的网格。

宽度比指定为 2:1,高度比设置为 1:2

subplot2grid 方法

subplot2grid 允许子图在网格索引中占用多个单元格。

def subplot2grid(shape,
                 loc,
                 rowspan=1,
                 colspan=1,
                 fig=None,
                 **kwargs):

shape 是网格的形状,具有 2 个整数的序列,其第一个元素是行数,第二个元素是列数。

locaxis 放置在网格中的位置。它也是 2 个整数的序列,其第一个元素是行号,第二个元素是列号。

rowspancolspanaxis 向右(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()

Matplotlib 中如何创建不同大小的子图 subplot

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) 中,并且占据其他两列。