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

Question

我想知道是否有任何选项可以使下图（sns.heatmap(df)子图的输出）变得平滑： 我刚刚在这里找到了一个相关的答案，这是通过使用建议的zsmooth：

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

我使用的代码片段seaborn如下：

#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()

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

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]})