fig, (ax1, ax2) = plt.subplots(1,2)，我想给子图加上颜色条

要为子图添加颜色条，你可以使用 `colorbar` 函数。在 `plt.subplots` 函数中，通过指定参数 `ax1` 和 `ax2`，我们创建了两个子图 `ax1` 和 `ax2`。然后，你可以在需要添加颜色条的子图上调用 `colorbar` 函数。下面是一个示例代码，展示了如何为两个子图添加颜色条：

import numpy as np
import matplotlib.pyplot as plt

# 生成随机数据
data1 = np.random.random((10, 10))
data2 = np.random.random((10, 10))

# 创建子图
fig, (ax1, ax2) = plt.subplots(1, 2)

# 绘制第一个子图
im1 = ax1.imshow(data1)
ax1.set_title('Subplot 1')

# 添加颜色条到第一个子图
cbar1 = fig.colorbar(im1, ax=ax1)

# 绘制第二个子图
im2 = ax2.imshow(data2)
ax2.set_title('Subplot 2')

# 添加颜色条到第二个子图
cbar2 = fig.colorbar(im2, ax=ax2)

# 设置整体布局
plt.tight_layout()

# 显示图形
plt.show()

在这个示例中，我们首先生成了两个随机数据矩阵 `data1` 和 `data2`。然后，我们使用 `plt.subplots` 函数创建了一个包含两个子图的图形窗口，并将其分配给变量 `fig` 和 `(ax1, ax2)`。接下来，我们在每个子图上绘制了热图，并使用 `set_title` 函数设置了子图的标题。然后，我们分别为每个子图调用了 `colorbar` 函数，并将其返回值分别赋值给变量 `cbar1` 和 `cbar2`。最后，我们使用 `tight_layout` 函数设置整体布局，并显示图形。

你可以根据需要调整代码，适应你的具体情况。希望对你有所帮助！