Plotly:如何使用按钮交互设置热图色标？

import plotly.graph_objects as go
from plotly.subplots import make_subplots

# 创建数据
z = [[1, 20, 30],
     [20, 1, 60],
     [30, 60, 1]]

# 创建热图对象
fig = go.Figure(data=go.Heatmap(z=z))

# 创建按钮选项列表
buttons = [
    dict(label='Option 1', method='update', args=[{'z': [[1, 20, 30], [20, 1, 60], [30, 60, 1]]}]),
    dict(label='Option 2', method='update', args=[{'z': [[1, 10, 50], [10, 1, 30], [50, 30, 1]]}]),
    dict(label='Option 3', method='update', args=[{'z': [[1, 40, 10], [40, 1, 20], [10, 20, 1]]}])
]

# 创建对应的色标值列表
colorscales = [
    [[0, 'rgb(255, 0, 0)'], [1, 'rgb(0, 255, 0)']],
    [[0, 'rgb(255, 255, 0)'], [1, 'rgb(0, 255, 255)']],
    [[0, 'rgb(0, 0, 255)'], [1, 'rgb(255, 0, 255)']]
]

fig.update_layout(
    updatemenus=[
        dict(
            buttons=buttons,
            direction="down",
            pad={"r": 10, "t": 10},
            showactive=True,
            x=0.1,
            xanchor="left",
            y=1.1,
            yanchor="top"
        ),
    ],
    coloraxis=dict(colorscale=colorscales[0])
)

fig.show()