如何根据自定义范围为 imshow 图表着色

Question

我正在制作一个用作热图的图表，但我想将特定颜色设置为 x 和 x+0.05 之间的值，而不是“传统”色标；

下图是一个很好的视觉示例，可以说明我想做什么：

我尝试了一些方法来做到这一点，但没有一种方法奏效。

下面是一个MWE

import pandas as pd
import plotly.express as px

dict_vals = [
            {"user1": None, "user2": 0.906, "user3": 0.842},
            {"user1": 0.906, "user2": None, "user3": 0.527},
            {"user1": 0.842, "user2": 0.527, "user3": None},
        ]

matrix_df=pd.DataFrame.from_dict(dict_vals)
fig = px.imshow(
    matrix_df,
    x=matrix_df.columns,
    y=matrix_df.columns,
    zmin=0.75,
    zmax=0.95,
    text_auto=True,
    aspect="auto"
)
colors = [
    "rgb(214, 11, 67)",
    "rgb(255, 107, 0)",
    "rgb(255, 201, 77)",
    "rgb(168, 227, 0)",
    "rgb(65, 175, 26)",
    ]

fig.update_coloraxes(
    showscale=False,
    colorscale=[
        (0.0, colors[0]),
        (0.8, colors[1]),
        (0.85, colors[2]),
        (0.9, colors[3]),
        (0.95, colors[4]),
        (1, colors[4]),
    ],
)

如果您对如何解决此问题有任何参考，我将不胜感激

Answer 1

我找到了一种方法来做我正在寻找的事情

import pandas as pd
import plotly.express as px
import numpy as np

dict_vals = [
            {"user1": None, "user2": 0.906, "user3": 0.842},
            {"user1": 0.906, "user2": None, "user3": 0.527},
            {"user1": 0.842, "user2": 0.527, "user3": None},
        ]

matrix_df=pd.DataFrame.from_dict(dict_vals)

color_list=[[0, colors[0]],
    [0.80, colors[0]], 
    [0.80, colors[1]],  
    [0.85, colors[1]],
    [0.85, colors[2]], 
    [0.90, colors[2]],
    [0.90, colors[3]], 
    [0.95, colors[3]],  
    [0.95, colors[4]],
    [1.00, colors[4]]]

fig = px.imshow(
    matrix_df.values,
    x=matrix_df.columns,
    y=matrix_df.columns,
    zmin=0,
    zmax=1,
    text_auto=True,
    aspect="auto",
    color_continuous_scale=color_list
)
fig.update_coloraxes(showscale=True)

结果将是：

我希望它可以帮助将来的人。

问候，莱昂纳多