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

时间:2018-10-20 08:30:52

标签: python colors plotly heatmap

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

但是在将图像打印在一张纸上之后,它看起来非常暗(彩色打印),并且在黑白打印后变得完全不可识别。我想知道是否可以任意更改颜色范围;这样整个图像看起来会更柔和或更亮。

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

注意:由于我已经问了另一个与密谋here有关的问题,因此不确定是否应该将其作为一个单独的问题。

1 个答案:

答案 0 :(得分: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]