python - plotly : Heatmap color legend i subplot

Question

我正在 python 中使用plotly，尝试创建两个热图子图，每个子图都有一个自定义图例。但我通过以下代码看到的是:

import plotly


fig = plotly.subplots.make_subplots(rows=2, cols=1)

mat = np.array([[1,0],
                [0,1]])


fig.add_trace(go.Heatmap(z=mat, colorscale='Bluered_r'), row=1, col=1)

fig.add_trace(go.Heatmap(z=mat*10, colorscale='Bluered_r'), row=2, col=1)

fig.show()

您可以看到两个图例都显示在同一位置。

感谢您的帮助!

Answer 1

刚刚使用 colorbar 参数找到了隐藏在文档中的解决方案:

import plotly


fig = plotly.subplots.make_subplots(rows=2, cols=1)

mat = np.array([[1,0],
                [0,1]])


fig.add_trace(go.Heatmap(z=mat, colorscale='Bluered_r',colorbar=dict(y=.8,len=.5)), row=1, col=1)

fig.add_trace(go.Heatmap(z=mat*10, colorscale='Bluered_r',colorbar=dict(y=.2,len=.5)), row=2, col=1)

fig.show()

这将给出: