如何为Seaborn热力图的Colorbar添加黑色边框（无需使用子图）

嘿，这个需求很好实现，完全不用折腾子图~ 其实Seaborn生成的热力图里，色条（colorbar）是可以直接通过返回的ax对象拿到的，然后我们只需要给它的边框设置样式就行。

直接看修改后的代码吧：

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

# 创建示例热力图
data = np.random.rand(10, 12)
ax = sns.heatmap(data)

# 获取对应的色条对象
cbar = ax.collections[0].colorbar
# 遍历色条的所有边框，设置为黑色并显示
for spine in cbar.spines.values():
    spine.set_edgecolor('black')
    spine.set_visible(True)

plt.show()

简单解释下步骤：

• ax.collections[0]是热力图的核心绘图元素，通过它的colorbar属性就能直接获取到我们要修改的色条对象
• 色条的边框是通过spines属性管理的，遍历所有边框后，设置颜色为黑色并让它们可见（默认有些边框可能是隐藏的）

如果只需要给特定的边框加黑（比如只加左右两侧），也可以单独指定边框来修改，比如：

cbar.spines['left'].set_edgecolor('black')
cbar.spines['left'].set_visible(True)
cbar.spines['right'].set_edgecolor('black')
cbar.spines['right'].set_visible(True)

这样就能轻松给色条加上黑色边框啦，完全不需要用到子图~

备注：内容来源于stack exchange，提问作者Caterina