如何为Seaborn热力图添加带不确定度的自定义标注

直接上解决方案：Seaborn的heatmap其实支持字符串类型的标注数组，你之前的问题大概率是没做好行列对齐的透视表，或者没正确设置fmt参数。下面是完整的实现步骤和代码：

步骤说明 #

• 分别对**数值列(D)和不确定度列(E)**做透视表，确保两者的行列结构完全一致
• 遍历两个透视表的单元格，拼接成"数值±不确定度"格式的字符串，生成标注用的NumPy数组
• 绘制热力图时，将annot设为这个字符串数组，同时指定fmt=''（避免Seaborn对字符串做数值格式化）

示例代码 #

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

# 构造示例数据（替换成你的真实数据）
data = pd.DataFrame({
    'Row': ['A', 'A', 'B', 'B'],
    'Col': ['X', 'Y', 'X', 'Y'],
    'D': [1.23, 4.56, 7.89, 2.34],
    'E': [0.12, 0.34, 0.56, 0.78]
})

# 分别生成数值和不确定度的透视表
pivot_values = pd.pivot_table(data, values='D', index='Row', columns='Col')
pivot_errors = pd.pivot_table(data, values='E', index='Row', columns='Col')

# 生成标注用的字符串数组
annot_array = np.array([
    [f"{val:.2f}±{err:.2f}" for val, err in zip(row_vals, row_errs)]
    for row_vals, row_errs in zip(pivot_values.values, pivot_errors.values)
])

# 绘制热力图
plt.figure(figsize=(8, 6))
sns.heatmap(
    pivot_values,
    annot=annot_array,
    fmt='',  # 关键：禁用数值格式化，直接输出字符串
    cmap='viridis',
    cbar=True,
    annot_kws={'size': 12}  # 可调整标注字体大小
)
plt.title('带不确定度标注的热力图')
plt.show()

关键注意点 #

1. 行列对齐：两个透视表必须使用相同的index和columns，否则字符串拼接时会出现错位
2. 字符串格式化：可以根据需求调整小数位数（比如.2f改成.1f），或者添加科学计数法格式
3. fmt参数：一定要设为''，否则Seaborn会尝试把字符串解析成数值，导致报错

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