如何在matplotlib中叠加两个热图？

import numpy as np
import matplotlib.pyplot as plt

data1 = np.random.rand(10, 10)  # 第一个热图的数据
data2 = np.random.rand(10, 10)  # 第二个热图的数据

fig, (ax1, ax2) = plt.subplots(1, 2)  # 创建一个1行2列的图形布局，并获取两个子图的对象

heatmap1 = ax1.imshow(data1, cmap='hot')  # 绘制第一个热图
ax1.set_title("Heatmap 1")  # 设置第一个热图的标题
fig.colorbar(heatmap1, ax=ax1)  # 添加第一个热图的颜色刻度条

heatmap2 = ax2.imshow(data2, cmap='hot')  # 绘制第二个热图
ax2.set_title("Heatmap 2")  # 设置第二个热图的标题
fig.colorbar(heatmap2, ax=ax2)  # 添加第二个热图的颜色刻度条

combined_heatmap = heatmap1.get_array() + heatmap2.get_array()  # 将两个热图的数据进行叠加
combined_heatmap = np.clip(combined_heatmap, 0, 1)  # 限制叠加后的数值范围在0-1之间
heatmap1.set_array(combined_heatmap)  # 更新第一个热图的数据为叠加后的数据

plt.show()