如何将Seaborn热图中指定列的标注转换为千(K)、百万(M)格式

我懂你的困扰：想把热图里指定列的标注改成「千(K)、百万(M)」的易读格式，但直接把列转成字符串后，Seaborn画热图就报错了——这是因为热图需要数值型数据来计算颜色映射，修改列类型后直接破坏了这个逻辑。

给你一个完美的解决方案：把颜色计算用的原始数值和显示用的格式化标注分开处理，具体步骤如下：

核心思路 #

Seaborn的heatmap的annot参数不仅可以设为True，还能接受一个二维数组。我们可以用原始数值数据生成热图的颜色梯度，同时单独生成一个包含格式化字符串的数组作为标注，这样既不影响颜色映射，又能显示你想要的K/M格式。

完整代码示例 #

import seaborn as sns
import matplotlib.pyplot as plt
from numerize import numerize as nz

# 假设你的原始数据是heatmap_df，'PropDmgAdj'是需要格式化的目标列
# 第一步：生成自定义标注矩阵
annot_df = heatmap_df.copy()
# 对目标列应用格式化，这里保留两位小数（和你示例里的格式匹配）
annot_df['PropDmgAdj'] = annot_df['PropDmgAdj'].apply(lambda x: nz.numerize(x, 2))
# 把数据框转换成二维数组，作为annot参数的输入
annot_matrix = annot_df.values

# 第二步：绘制热图
sns.heatmap(heatmap_df, 
            cmap='rocket_r', 
            annot=annot_matrix,  # 使用我们自定义的标注数组
            fmt='',  # 关键！因为标注是字符串，所以不要设置数值格式
            linewidths=2, 
            cbar_kws={'label': 'Percentiles', 'orientation': 'vertical'}, 
            vmin=0.95, 
            vmax=1, 
            xticklabels=xticks_label)

plt.show()

为什么之前会报错？ #

你之前直接修改了heatmap_df里的列类型为字符串，Seaborn绘制热图时需要基于数值数据计算颜色梯度，混合了字符串的DataFrame会让它无法正常解析，所以抛出错误。现在的方法完全分离了数据层和显示层，完美避开了这个问题。

额外小技巧 #

如果需要对多列或者整个热图的标注都做K/M格式化，只需要在annot_df里批量处理对应的列就行，比如用applymap对整个数据框处理：

annot_df = annot_df.applymap(lambda x: nz.numerize(x, 2))

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