python - 在 Matplotlib 中定位 5 个子图

Question

我想放置 5 个子图，其中顶部有 3 个，底部有 2 个，但彼此相邻。当前代码接近，但我希望最终结果如下所示(忽略灰线):

import matplotlib.pyplot as plt

ax1 = plt.subplot(231)
ax2 = plt.subplot(232)
ax3 = plt.subplot(233)
ax4 = plt.subplot(234)
ax5 = plt.subplot(236)

plt.show()

Answer 1

当你使用 suplot2grid 而不是 subplot 时，你可以使用 colspan。

import matplotlib.pyplot as plt

ax1 = plt.subplot2grid(shape=(2,6), loc=(0,0), colspan=2)
ax2 = plt.subplot2grid((2,6), (0,2), colspan=2)
ax3 = plt.subplot2grid((2,6), (0,4), colspan=2)
ax4 = plt.subplot2grid((2,6), (1,1), colspan=2)
ax5 = plt.subplot2grid((2,6), (1,3), colspan=2)

然后每个子图需要 2 列宽，以便第二行中的子图可以移动 1 列。