绘制大型 NetworkX 图时返回进度状态

Question

• 我正在绘制一个需要很长时间才能绘制的大图 过程。
• 是否可以返回绘图当前状态的status、current_node 或percentage？
• 我不打算逐步绘制网络，因为我所做的只是将其保存为高 dpi 图像。

这是我正在使用的代码示例：

path = nx.shortest_path(G, source=u'1234', target=u'98765')
path_edges = zip(path, path[1:])
pos = nx.spring_layout(G) 
nx.draw_networkx_nodes(G,pos,nodelist=path,node_color='r') 
nx.draw_networkx_edges(G,pos,edgelist=path_edges,edge_color='r',width=10) 
plt.axis('equal')
plt.savefig('prototype_map.png', dpi=1000)
plt.show()

Answer 1

我相信这样做的唯一方法是容纳绘制函数的源代码以打印 10%、20% 完成的内容....但是当我检查 source code of draw_networkx_nodes & draw_networkx 时，我意识到这不是直截了当的任务，因为 draw 函数将位置（节点和边）存储在一个 numpy 数组中，将其发送到 matplotlib (sourcecode) 的 ax.scatter 函数，这在不弄乱某些东西的情况下有点难以操作。我唯一能想到的就是改变：

xy = numpy.asarray([pos[v] for v in nodelist]) # In draw_networkx_nodes function

到

xy = []
count = 0
for v in nodelist:
    xy.append(pos[v])
    count +=1
    if (count == len(nodelist)):
        print '50% of nodes completed'
print '100% of nodes completed'
xy = numpy.asarray(xy)

类似地，当调用 draw_network_edges 时，指示边缘绘制的进度。我不确定这与事实相差多远，因为我不知道在 ax.scatter 函数中花费了多少时间。我还查看了 scatter 函数的源代码，但我无法确定循环或其他内容来打印已完成某些进展的指示。