python - 在 Plotly 中更改热图的配色方案

Question

我想使用 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')

这给了我这样的图像：

但是将图像打印在一张纸上后，它看起来很暗（彩色打印），并且在黑白打印后变得完全无法辨认。我想知道是否可以任意更改颜色范围；使整个图像看起来更柔和或更轻。

例如：1 可以是白色，0 可以是非常浅的蓝色或更柔和的颜色；使整个图像在纸张上打印后看起来更亮。

注意：不确定这是否应该是一个单独的问题，因为我在这里提出了另一个与 plotly 相关的问题。

标签： pythoncolorsplotlyheatmap

Answer 1

您将需要手动插入色标

trace = go.Heatmap(z=probability_matrix, colorscale=[[0.0, '#F5FFFA'], 
                         [0.2, '#ADD8E6'], 
                         [0.4, '#87CEEB'],
                         [0.6, '#87CEFA'], 
                         [0.8, '#40E0D0'], 
                         [1.0, '#00CED1']])
data=[trace]