Plotly：如何更改热图中颜色条的刻度文本

Question

我正在使用 Plotly 包进行可视化。
此 MWE 创建一个 10x10 矩阵并将其绘制在热图中。

import numpy as np
import pandas as pd
import plotly.graph_objects as go

vals = np.random.rand(10,10)*5
vals = np.around(vals)

tick_text = ['A','B','C','D','E']

df = pd.DataFrame(vals)

fig = go.Figure(go.Heatmap(
    z=df.values,
    x = df.index,
    y = df.columns
))
fig.update_layout(
    height=500,
    width=500
)
fig.update_layout(
            coloraxis_colorbar=dict(
                tickmode = 'array',
                tickvals = [i for i in range(5)],
                ticktext = tick_text,
            )
            )
fig.show()

这是输出

我正在尝试更改颜色栏旁边显示的文本。在本例中，我想看到 A、B、C、D、E，而不是 1、2、3、4、5。
我遵循了文档，据我所知，我必须更改

coloraxis_colorbar

，但这似乎不起作用。

Answer 1

我认为你想要的是

update_traces()

使用

labelalias

：

fig.update_traces(
    colorbar_labelalias={i: j for i, j in zip(range(5), tick_text)},
    selector=dict(type='heatmap')
)