在图片上绘制 seaborn 热力图代码

# 定义从白色到黑色的颜色梯度
colors = ["white", "black"]
cmap_name = "white_black"
n_bins = 4  # 定义梯度中的颜色数量

# 创建LinearSegmentedColormap对象
white_black_cmap = LinearSegmentedColormap.from_list(cmap_name, colors, N=n_bins)

# 假设你有一张128x128的图像文件
image_path = f"save/im{i}.png"  # 替换为你的图像文件路径
image = Image.open(image_path)
# image = image.resize((128, 128))  # 确保图像是128x128大小

data = 生成一个和 image 一样大的数据, 如 128x128

# 设置图形的风格为白色背景
sns.set_theme(style="white")

# 创建一个matplotlib的图和坐标轴对象
fig, ax = plt.subplots(figsize=(10, 8))

# 首先绘制底层图像
ax.imshow(image, extent=[0, 128, 0, 128], origin="lower")

# 然后叠加热力图，通过设置alpha参数来调整透明度
sns.heatmap(data, cmap=white_black_cmap, alpha=0.6, ax=ax, cbar=False, zorder=2)

# 移除坐标轴的刻度标签
ax.set_xticks([])
ax.set_yticks([])

# 设置标题
# ax.set_title('Overlay Heatmap on Image', fontsize=20, pad=20)

# 显示图形
# plt.show()
plt.savefig(f"save/hwadd{i}.pdf", format="pdf", bbox_inches="tight", pad_inches=0.0)

效果如下，上边的 cmap 可以改，也提供了自定义cmap的部分