Сглаживание тепловой карты в plotly

Я хотел создать тепловую карту матрицы плотности вероятности, используя plotly.

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

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

basic-heatmap

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


person Ahsan Tarique    schedule 29.04.2018    source источник


Ответы (1)


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

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

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

введите описание изображения здесь

person byouness    schedule 30.04.2018