如何更改 plotly 热图中的颜色

Question

我有以下示例dataframe矩阵，下面是我使用创建的一些函数生成的：

             Loan_ID    Gender   Married     Dependents   Education
Loan_ID      1.000       NaN      NaN          0.000        0.000
Gender        NaN       1.000     NaN           NaN          NaN
Married      0.638      0.638    1.000         0.638        0.638
Dependents   0.000      0.000    0.000         1.000        0.000
Education    0.502      0.502    0.502         0.502        1.000

我正在尝试使用plotly到 plot 热图，但使用基于某些值的特定颜色。 基于 dataframe。 如果值小于 0.05，我希望单元格为绿色如果单元格大于 0.05 但小于 0.1，我希望颜色为绿色。 条件语句如下所示：

colorscales = []
data_mask = df_mask.to_numpy()

for row in data_mask:
    for value in row:
        if np.isnan(value):
            color = "#f8fffa"
        elif float(value) < 0.05:
            color = "#10c13b"
        elif (float(value) > 0.05) and (float(value) < 0.1):
            color = '#fac511'
        colorscales.append(color)

我希望 plotly 热图中显示的颜色被这些颜色反映。 我尝试在图形布局中使用色阶和 bgcolor，但没有任何效果。 任何建议将不胜感激

Answer 1

我通过使用带有 colorscale argument的比例 [0.1] 解决了这个问题，如下所示：

colorscale = [[0, "#10c13b"],
             [0.05, '#10c13b'],
             [0.051, '#fac511'],
             [0.1, "#fac511"],
             [0.11, "#f71907"],
             [0.2, "#f71907"],
              [0.3, "#f71907"],
              [0.4, "#f71907"],
              [0.5, "#f71907"],
              [0.6, "#f71907"],
              [0.7, "#f71907"],
              [0.8, "#f71907"],
              [0.9, "#f71907"],
              [1.0, "#f71907"]]

这有效地解决了我的问题，以防万一有人需要这样做！