在 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相关的问题。

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]