在情节上平滑热图

Question

我想用图表创建概率密度矩阵的热图。

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

probability_matrix = np.loadtxt("/path/to/file")
trace = go.Heatmap(z = probability_matrix)
data=[trace]
plot(data, filename='basic-heatmap')

这给了我一个像这样的图像：

我想平滑正方形的颜色，以便图像中相邻正方形之间的过渡有点“平滑”＃34;。我想知道是否有办法做到这一点，而无需使用插值手动调整矩阵大小。

Answer 1

您可以使用zsmooth参数，该参数可以包含三个值（'fast'，'best'或False）。例如：

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

将为您提供以下平滑热图：