colorbar单独设置一个轴对象，再对轴对象进行灵活设置

import numpy as np
import matplotlib.pyplot as plt

# 创建一个二维随机数组
data = np.random.rand(10, 10)

# 创建一个图形和一个子图
fig, ax = plt.subplots()

# 绘制热力图
heatmap = ax.imshow(data, cmap='hot')
# print(ax.get_position().height)
# 添加 colorbar 的轴对象
cax = fig.add_axes([0.92, 0.1, 0.02, ax.get_position().height])  # [left, bottom, width, height]

# 创建 colorbar，并将长度设置为与热力图的y轴高度一致
colorbar = fig.colorbar(heatmap, cax=cax)

# 显示图形
plt.show()