plt.subplots.colorbar的gridspec

`plt.subplots()` 是一个方便的函数，可以在单个 `figure` 中创建多个子图。`gridspec` 参数允许您指定子图的布局。`gridspec` 可以是一个元组 `(rows, cols)`，也可以是一个长度为 3 的列表或元组，其中包含 `(rows, cols, subplot_spec)`。 `subplot_spec` 是 `gridspec` 对象，可用于更详细地定义子图的位置和大小。

`colorbar` 参数指示是否要在图的侧面添加颜色条。如果设置为 True，则在 `subplot_spec` 中指定的子图区域中的图形添加颜色条。

例如，以下代码创建了一个具有 2x2 布局的子图，并在第一个子图中绘制了一个热图，然后在第二个子图中绘制了与第一个子图相同的热图，但添加了一个颜色条。

import matplotlib.pyplot as plt
import numpy as np

# 创建一个 2x2 的子图布局
gs = plt.GridSpec(2, 2)

# 在第一个子图中绘制热图
ax1 = plt.subplot(gs[0, 0])
data = np.random.rand(10, 10)
im = ax1.imshow(data, cmap='viridis')
plt.colorbar(im)

# 在第二个子图中绘制与第一个子图相同的热图，并添加颜色条
ax2 = plt.subplot(gs[0, 1])
im = ax2.imshow(data, cmap='viridis')
plt.colorbar(im)

# 在第三个子图中绘制柱状图
ax3 = plt.subplot(gs[1, :])
x = np.arange(10)
y = np.random.rand(10)
ax3.bar(x, y)

plt.show()