为 python networkx 图中的节点添加工具提示

Question

我使用 networkx.DiGraph 创建了一个有向图，然后使用 networkx.draw_spring(graph) 绘制了它，因此图中的所有节点都有一些详细信息存储在字典列表中。

如何添加类似 "tooltip" 的内容以在鼠标悬停在每个节点上时查看这些详细信息？如果可能的话，如何使这个 "tooltip" 对所有节点始终可见，而不仅仅是通过悬停？

Answer 1

始终可见

要标记所有节点，只需要使用annotate。像这样

import matplotlib.pyplot as plt
import networkx as nx

G = nx.path_graph(5)
attrs = {0: {'attr1': 20, 'attr2': 'nothing'}, 1: {'attr2': 3}, 2: {'attr1': 42}, 3: {'attr3': 'hello'}, 4: {'attr1': 54, 'attr3': '33'}}
nx.set_node_attributes(G, attrs)
nx.draw(G)

for node in G.nodes:
    xy = pos[node]
    annot.xy = xy
    node_attr = G.nodes[node]
    text = '\n'.join(f'{k}: {v}' for k, v in G.nodes[node].items())
    text = f'node {node}\n' + text
    ax.annotate(text, xy=xy)

悬停

这是一个在悬停时获取工具提示的工作示例。这是基于使用标准 matplotlib 图 here 的工具提示。我使用 draw_networkx_nodes 来获取用于悬停和显示工具提示的对象，而不是使用 draw_spring。但是你可以手动定义位置 spring_layout.

import matplotlib.pyplot as plt
import networkx as nx

G = nx.path_graph(5)
attrs = {0: {'attr1': 20, 'attr2': 'nothing'}, 1: {'attr2': 3}, 2: {'attr1': 42}, 3: {'attr3': 'hello'}, 4: {'attr1': 54, 'attr3': '33'}}
nx.set_node_attributes(G, attrs)

fig, ax = plt.subplots()
pos = nx.spring_layout(G)
nodes = nx.draw_networkx_nodes(G, pos=pos, ax=ax)
nx.draw_networkx_edges(G, pos=pos, ax=ax)

annot = ax.annotate("", xy=(0,0), xytext=(20,20),textcoords="offset points",
                    bbox=dict(boxstyle="round", fc="w"),
                    arrowprops=dict(arrowstyle="->"))
annot.set_visible(False)

def update_annot(ind):
    node = ind["ind"][0]
    xy = pos[node]
    annot.xy = xy
    node_attr = {'node': node}
    node_attr.update(G.nodes[node])
    text = '\n'.join(f'{k}: {v}' for k, v in node_attr.items())
    annot.set_text(text)

def hover(event):
    vis = annot.get_visible()
    if event.inaxes == ax:
        cont, ind = nodes.contains(event)
        if cont:
            update_annot(ind)
            annot.set_visible(True)
            fig.canvas.draw_idle()
        else:
            if vis:
                annot.set_visible(False)
                fig.canvas.draw_idle()

fig.canvas.mpl_connect("motion_notify_event", hover)

plt.show()

Answer 2

仅供参考@busybear @sudofix

这仅在您的节点从 0 开始时有效。

如果你这样做：

import matplotlib.pyplot as plt
import networkx as nx

nodes = list(range(5))

edges = []
for e1,e2 in zip(nodes[:-1],nodes[1:]):
    edges.append((e1,e2))

G = nx.Graph()
G.add_nodes_from(nodes)
G.add_edges_from(edges)

attrs = {}
for node in G.nodes:
    attrs[node] = {'attr1': node, 'attr2': 'hello', 'attr3': 33}

剩下的保持不变

nx.set_node_attributes(G, attrs)

fig, ax = plt.subplots()
pos = nx.spring_layout(G)
nodes = nx.draw_networkx_nodes(G, pos=pos, ax=ax)
nx.draw_networkx_edges(G, pos=pos, ax=ax)

annot = ax.annotate("", xy=(0,0), xytext=(20,20),textcoords="offset points",
                    bbox=dict(boxstyle="round", fc="w"),
                    arrowprops=dict(arrowstyle="->"))
annot.set_visible(False)

def update_annot(ind):
    node = ind["ind"][0]
    xy = pos[node]
    annot.xy = xy
    node_attr = {'node': node}
    node_attr.update(G.nodes[node])
    text = '\n'.join(f'{k}: {v}' for k, v in node_attr.items())
    annot.set_text(text)

def hover(event):
    vis = annot.get_visible()
    if event.inaxes == ax:
        cont, ind = nodes.contains(event)
        if cont:
            update_annot(ind)
            annot.set_visible(True)
            fig.canvas.draw_idle()
        else:
            if vis:
                annot.set_visible(False)
                fig.canvas.draw_idle()

fig.canvas.mpl_connect("motion_notify_event", hover)

plt.show()

一切正常。

但是如果你改变 nodes = list(range(5)) 成为 nodes = list(range(1,6)) 它不起作用，因为 node = ind["ind"][0] returns 节点在 G.nodes 中的位置而不是节点的名称，因此访问 pos[node] 和 G.nodes[node] 得到位置错误（移动 1）。

解决方案是创建一个像

这样的映射

idx_to_node_dict = {}
for idx, node in enumerate(G.nodes):
    idx_to_node_dict[idx] = node

并修复函数以将其用作：

def update_annot(ind):
    node_idx = ind["ind"][0]
    node = idx_to_node_dict[node_idx]

为 python networkx 图中的节点添加工具提示

adding tooltip for nodes in python networkx graph

python

graph

matplotlib

tooltip

networkx

始终可见

悬停