如何修复networkx中的节点position？

Question

我最近开始在 python 中使用 Networkx。 我有一个代码，它生成一个网络，并根据一些过程节点功能发生变化。 例如颜色和状态。 为了绘制图表，我使用以下 function

def draw_graph():
    colors = []
    for i in range (nCount):
        for j in range (i,nCount):
            if ifActive(i,j,timeStep) == 1 :
                
                    colors.append('r')
                  
                
            else :
                colors.append('g')
    color_map = []   
    nColor= nx.get_node_attributes(graph,'color')
    for nc in nColor:
        color_map.append(nColor[nc])   
    nx.draw(graph,pos=nx.spring_layout(graph), node_color = color_map, edge_color = colors,with_labels = True )

并在for循环中的主要function中，我将绘图称为function，但每次position的节点都会发生变化。 现在我想知道，有没有办法修复所有图纸中节点的position？ 如果是，我该怎么做？ 这是主要的 function

draw_graph()
for time in range(1,timeStep+1):
         if graph.node[i]["status"] == 1:
                settime(time,i)
        plt.figure(figsize=[10, 10], dpi=50)
        draw_graph()

下图是output的例子。 如果您根据标签考虑节点，则它们的 position 不固定。

Answer 1

正如@furas 在评论中所述，为了始终获得相同的节点位置，您需要将其保留为变量，例如：

pos = nx.spring_layout(graph)

然后将您的图表绘制为：

def draw_graph(p):
    # any code here, as long as no further nodes are added.
    nx.draw(graph,pos=p)

然后您最终可以将其称为：

pos = nx.spring_layout(graph)
draw_graph(pos)
# any code here, as long as no further nodes are added.
draw_graph(pos)
# each call will give the same positions.