Python Netgraph 交互式标签

Python Netgraph Interactive Labels

我正在尝试使用 netgraph 和 networkx 创建交互式绘图。

我希望绘图允许节点移动,并且随着节点移动,边缘和 edge_labels 也会动态更新。

移动节点已由 netgraph here 的作者解决。现在,当我制作一个更简单的图并尝试标记边缘时,标签保持静态,有时甚至不在边缘上。

看起来处理 edge_positions 类似于最后两行的 node_positions 应该至少解决标签不动的问题。为什么标签没有锚定到特定边缘仍然令人费解。有谁知道能不能达到想要的效果?

这是移动任何东西之前的片段:

这是将右下角的节点移动到左上角后的截图:

这是我当前的代码:

   
import matplotlib.pyplot as plt
import networkx as nx
import netgraph # pip install netgraph

#Graph creation:
G=nx.Graph(type="")

for i in range(6):
     G.add_node(i,shape="o")

#Changing shape for two nodes
G.nodes[1]['shape'] = "v"
G.nodes[5]['shape'] = "v"

#Add edges
G.add_edge(1,2)
G.add_edge(4,5)
G.add_edge(0,4)
G.add_edge(2,3)
G.add_edge(2,4)

labs={(1,2):"1 to 2"}

nx.draw_networkx_edge_labels(G, pos=nx.spring_layout(G),edge_labels=labs)
#Get node shapes
node_shapes = nx.get_node_attributes(G,"shape")


# Create an interactive plot.
# NOTE: you must retain a reference to the object instance!
# Otherwise the whole thing will be garbage collected after the initial draw
# and you won't be able to move the plot elements around.

pos = nx.layout.spring_layout(G)

######## drag nodes around #########

# To access the new node positions:
plot_instance =   netgraph.InteractiveGraph(G, node_shape=node_shapes, node_positions=pos, edge_positions=pos)


node_positions = plot_instance.node_positions
edge_positions = plot_instance.edge_positions

要将其作为正式答案,您需要向 InteractiveGraph 对象添加要绘制(和移动)边缘标签的信息,即以下

netgraph.InteractiveGraph(G, 
                          node_shape=node_shapes, 
                          node_positions=pos, 
                          edge_positions=pos, 
                          edge_labels=labs)

(重点强调最后一个参数)

正如您已经注意到的那样,您不需要 nx.draw_networkx_edge_labels 的调用。