Python Seaborn 热图可视化及保存方法

你可以使用 plt.savefig() 方法将生成的热图保存为图片格式。以下是保存为 png 格式的示例代码：

import matplotlib.pyplot as plt
import seaborn as sns

plt.subplots(figsize=(9, 9))
plt.rcParams['font.sans-serif'] = ['FangSong']
plt.rcParams['axes.unicode_minus'] = False
plt.title('变量相关系数 - 热图\n', fontsize=18)
sns.heatmap(corr, annot=True, vmax=1, vmin=0, xticklabels=True, yticklabels=True, square=True, cmap="Reds",
            linewidths=0.05, linecolor='white')
plt.savefig('heatmap.png', dpi=300)  # 保存为 png 格式，dpi 参数设置图片分辨率
plt.show()

这样会将热图保存为当前工作目录下的 heatmap.png 文件。你可以根据需要修改文件名和保存路径。