如何在 Seaborn 中轻松平滑热图？

Question

我想知道是否有任何选项可以制作以下图片，这些图片是sns.heatmap(df) subplots smoothy的输出： 我刚刚找到了一个相关答案here，建议使用zsmooth：

data = [go.Heatmap(z=[[1, 20, 30],
                      [20, 1, 60],
                      [30, 60, 1]],
                   zsmooth = 'best')]
iplot(data)

我使用seaborn的代码的sn-p如下：

#plotting all columns ['A','B','C'] in-one-window side by side
fig, axes = plt.subplots(nrows=1, ncols=3 , figsize=(20,10))

plt.subplot(131)
sns.heatmap(df1, vmin=-1, vmax=1, cmap ="coolwarm", linewidths=.75 , linecolor='black', cbar=True , cbar_kws={"ticks":[-1.0,-0.75,-0.5,-0.25,0.00,0.25,0.5,0.75,1.0]})
fig.axes[-1].set_ylabel('[MPa]', size=20) #cbar_kws={'label': 'Celsius'}
plt.title('A', fontsize=12, color='black', loc='center', style='italic')
plt.axis('off')

plt.subplot(132)
sns.heatmap(df2, vmin=-1, vmax=1, cmap ="coolwarm", linewidths=.75 , linecolor='black', cbar=True , cbar_kws={"ticks":[-1.0,-0.75,-0.5,-0.25,0.00,0.25,0.5,0.75,1.0]})
fig.axes[-1].set_ylabel('[Mpa]', size=20) #cbar_kws={'label': 'Celsius'}
plt.title('B', fontsize=12, color='black', loc='center', style='italic')
plt.axis('off')

plt.subplot(133)
sns.heatmap(df3, vmin=-40, vmax=150, cmap ="coolwarm" , linewidths=.75 , linecolor='black', cbar=True , cbar_kws={"ticks":[-40,150,-20,0,25,50,75,100,125]}) 
plt.title('C', fontsize=12, color='black', loc='center', style='italic')
plt.axis('off')


plt.suptitle(f'Analysis of data in cycle Nr.: {count}', color='yellow', backgroundcolor='black', fontsize=48, fontweight='bold')
plt.subplots_adjust(top=0.7, bottom=0.3, left=0.05, right=0.95, hspace=0.2, wspace=0.2)
plt.savefig(f'{i}/{i}{i}{count}.png') 
plt.show()

问题是我不确定我是否可以使用它，因为它调用以下库，而我的是另一个库。如果有人向我解释是否可能以及如何在我的 sn-p 上实现它，那就太好了？

from plotly.offline import download_plotlyjs, init_notebook_mode, plot
import plotly.graph_objs as go

Answer 1

您链接的问题使用情节。如果您不想使用它并且只想平滑数据的外观，我建议您只使用使用 scipy 的高斯滤波器。

在顶部，导入

from scipy.ndimage.filters import gaussian_filter

然后像这样使用它：

df3_smooth = gaussian_filter(df3, sigma=1)
sns.heatmap(df3_smooth, vmin=-40, vmax=150, cmap ="coolwarm" , cbar=True , cbar_kws={"ticks":[-40,150,-20,0,25,50,75,100,125]}) 

您可以更改平滑量，例如使用sigma=3，或任何其他可以提供所需平滑量的数字。

请记住，这也会“消除”您拥有的任何最大数据峰值，因此您的最小和最大数据将不再与您在规范化中指定的相同。为了仍然获得好看的热图，我建议不要为您的 vmin 和 vmax 使用固定值，但是：

sns.heatmap(df3_smooth, vmin=np.min(df3_smooth), vmax=np.max(df3_smooth), cmap ="coolwarm" , cbar=True , cbar_kws={"ticks":[-40,150,-20,0,25,50,75,100,125]})