Matplotlib 中如何建立不同大小的子圖 subplot
Jinku Hu
2023年1月30日
我們可以使用 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
