Seaborn热图不断在数据集上循环添加颜色条[重复]

Question

我正在尝试在各种数据集上循环绘制热图，但是对于每个新的热图，都会添加一个新的颜色条（地图看起来很好，并且没有添加额外的地图）。我可以通过重置循环内的颜色条来使用解决方法，但我更愿意知道发生了什么并有一个更清洁的解决方案。提前感谢您的帮助！

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

# read file
atlas = ['A','B','C','D']
output_path = '/Users/polo/Desktop/Heatmaps/'

for at in range(len(atlas)):
    data = pd.read_csv('/Users/polo/Desktop/correl_input_{}.csv'.format(atlas[at]))
    hmap = sns.heatmap(data,cmap='seismic',linewidths=.5,vmin=-0.1, vmax=0.1)
    hmap.set_ylim(0, 5)
    plt.savefig(output_path + 'Heatmap_{}.png'.format(atlas[at]), dpi=1200,bbox_inches='tight')
    #plt.savefig('{}_plot.png', format='png', dpi=1200,bbox_inches='tight')

Answer 1

TL;DR 您的plt.figure() 解决方案是正确的，正如here 所建议的那样。

seaborn 建立在 matplotlib 之上，使用图形和轴的概念。创建matplotlib.pyplot 图表的规范方法从使用f, ax = plt.subplots() 实例化图形和轴开始。当您调用坐标轴级别的绘图（例如heatmap、seaborn calls matplotlib.pyplot.gca()）时，它会获取当前坐标轴。如果它不存在，seaborn 会在后台实例化一个新的。

我猜想在您的循环中，热图相互覆盖，但 seaborn 正在动态调整图形以为每个颜色条留出空间。用plt.figure()（或f, ax = plt.subplots()）清除数字就是你想要的。