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

Change color scheme of heatmap in 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')

这给了我这样的图像:

但是打印在一张纸上之后,看起来很暗(彩色打印),黑白打印后就完全看不清了。我想知道我是否可以任意更改颜色范围;使整个图像看起来更柔和或更轻。

例如:1 可以是白色,0 可以是非常 的蓝色或更柔和的颜色;这样在纸上打印后整个图像看起来更亮。

注意:不确定这是否应该是一个单独的问题,因为我问了另一个与 plotly 有关的问题

您需要手动插入色阶

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]