使用seaborn库批量生成图像并保存

场景：用python的seaborn库批量绘制热度图并保存

遇到的问题：seaborn绘制的多幅图像重叠在一起

解决办法：在保存图像的代码后加上plt.close()

import seaborn as sns
from matplotlib import pyplot as plt
#for循环内
heatmap= sns.heatmap(data, cmap="YlGnBu_r", xticklabels=False, yticklabels=False)#调用seaborn绘图
heatmap.get_figure().savefig(os.path.join(save_path,file_name)#保存图像
plt.close()