如何将第二个数据集添加到绘图注释的热图中？

Question

我正在尝试创建一个带注释的热图，该热图带有一个下拉菜单，可以在两组不同的数据之间切换。数据集具有相同的格式，并且我添加了一个有效的下拉菜单。但我一次只能添加一个数据集。我正在使用

fig = ff.create_annotated_heatmap(data, annotation_text=numbers, showscale=True, colorscale=colorscale, text=hover, hoverinfo='text')

创建带注释的热图。有没有办法添加第二个数据集，以便使用下拉菜单进行切换？

Answer 1

已解决。我必须将第二个数据集添加到下拉菜单对象的args中，以及所需的任何其他更改(例如悬停文本)

Answer 2

我刚刚意识到使用菜单在两个绘图之间切换是多么容易。您只需从每个图中获取数据，就可以创建一个跟踪列表，以便在

from plotly.offline import init_notebook_mode, iplot
import plotly.figure_factory as ff
init_notebook_mode(connected=True)

fig_1 = ff.create_annotated_heatmap(...)
fig_2 = ff.create_annotated_heatmap(...)
menu_items = ["Heatmap 1", "Heatmap 2"]

trace1 = fig_1.to_dict()["data"][0]
trace2 = fig_2.to_dict()["data"][0]

buttons = []
for i, menu_item in enumerate(menu_items):
    visibility = [i==j for j in range(len(menu_items))]
    button = dict(
                 label  =  menu_item,
                 method = 'update',
                 args   = [{'visible': visibility},
                           { 'title' : menu_item }])
    buttons.append(button)

updatemenus = list([
    dict(buttons = buttons)
])
layout = dict(updatemenus = updatemenus, title=menu_items[0])
fig = dict(data=[trace1, trace2], layout=layout)
iplot(fig)