如何在Python中为热力图添加初始与边界条件的星型标记

如果你用matplotlib/seaborn绘制热力图，只需在热力图绘制完成后，通过散点图函数添加星型标记即可，具体操作如下：

1. 准备标记点坐标 #

先整理好需要标注的初始/边界数据点的x、y坐标：

• 若热力图基于矩阵索引，x对应列索引，y对应行索引
• 若用自定义坐标轴，需确保标记点坐标与刻度值匹配

2. 绘制基础热力图 #

以seaborn为例，先画出你的热力图：

import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np

# 示例数据，替换成你的真实数据
heatmap_data = np.random.rand(8, 8)
ax = sns.heatmap(heatmap_data, cmap='coolwarm', annot=False)

3. 添加星型标记 #

使用plt.scatter()，通过marker='*'指定星型样式，同时调整参数保证标记醒目：

# 示例标记点，替换成你的目标点坐标
target_x = [0, 7, 0, 7]  # 列索引
target_y = [0, 0, 7, 7]  # 行索引

# 绘制星型标记
plt.scatter(target_x, target_y, 
            marker='*', 
            s=250,        # 标记大小
            c='yellow',   # 填充色
            edgecolor='black',  # 边缘色，避免和热力图融合
            zorder=10)    # 图层优先级，确保标记在最上层

4. 自定义调整（可选） #

如果需要适配自定义坐标轴刻度，只需将标记点坐标转换为对应刻度值：

# 示例：自定义x轴刻度
custom_x = np.linspace(0, 100, 8)
ax.set_xticks(np.arange(8))
ax.set_xticklabels([f'{x:.0f}' for x in custom_x])
# 此时标记点x坐标需对应custom_x的位置，或直接用索引映射

最后执行plt.show()就能得到带星型标记的热力图。

内容的提问来源于stack exchange，提问作者Simone Liu