python热力图上添加注释

在Python中，你可以使用seaborn库创建热力图，并通过matplotlib库添加注释。以下是基本步骤：

首先，安装必要的库：

pip install seaborn matplotlib

然后，创建一个简单的热力图示例：

import seaborn as sns
import numpy as np

# 创建数据矩阵
data = np.random.rand(5, 5)

# 使用seaborn绘制热力图
heatmap = sns.heatmap(data, annot=True, fmt=".2f") # "annot=True" 表示显示数值，".2f" 控制小数位数

# 添加自定义注释
for i in range(len(heatmap.axes.flat)):
    text = heatmap.texts[i].get_text()  # 获取文本
    x, y = heatmap.axes.flat[i].get_position().x0 + .02, heatmap.axes.flat[i].get_position().y0 + .02  # 设置注释位置偏移
    heatmap.text(x, y, text, color='black', ha='left', va='bottom')  # 添加注释

# 显示图形
plt.show()

在这个例子中，`sns.heatmap()`函数用于创建热力图，`annot=True`表示添加数值标签，`.2f`控制精度到小数点后两位。之后，遍历每个单元格，获取其文本内容并添加自定义位置的注释。