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

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

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

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