Сглаживать тепловую карту на графике

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')

Это дает мне такое изображение:

Я хочу сгладить цвет квадратов, чтобы переход между соседними квадратами на изображении был несколько «более плавным». Мне было интересно, есть ли способ сделать это, не изменяя размер матрицы вручную с помощью интерполяции.

Answer 1

Вы можете использовать аргумент zsmooth, который может принимать три значения ('fast', 'best' или False). Например:

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

Даст вам следующую гладкую тепловую карту: