非常缓慢的网络图动画在Python的matplotlib与networkx

我打算动画一个networkx图随着时间的推移而增长。我有它与弯曲的边缘和所有工作。非常高兴，但是随着图表的增长，它变慢了！任何人都可以解释或更改给定的代码以更快运行。我认为这不是我的电脑问题。注：即使没有花式弯曲的边缘，它非常缓慢。一个简单的nx.draw会导致同样的问题非常缓慢的网络图动画在Python的matplotlib与networkx

import time 
from matplotlib import pyplot as plt 
from matplotlib import animation 
import networkx as nx 
from matplotlib.patches import FancyArrowPatch, Circle 
import numpy as np 

mainGraph = nx.DiGraph() 
nodeToCoordinate_dict = {} 
label_coordinatePositions = {} 
edgeLabel_coordinatePositions = {} 
edgeLabel_values_dict = {} 
fig = plt.figure() 


#============================================================= 

def draw_network(G,pos,ax,sg=None): 
    e = None 
    for n in G: 
     c=Circle(pos[n],radius=0.02,alpha=0.5) 
     ax.add_patch(c) 
     G.node[n]['patch']=c 
     x,y=pos[n] 
    seen={} 
    for (u,v,d) in G.edges(data=True): 
     n1=G.node[u]['patch'] 
     n2=G.node[v]['patch'] 
     rad=0.1 
     if (u,v) in seen: 
      rad=seen.get((u,v)) 
      rad=(rad+np.sign(rad)*0.1)*-1 
     alpha=0.5 
     color='k' 

     e = FancyArrowPatch(n1.center,n2.center,patchA=n1,patchB=n2, 
          arrowstyle='-|>', 
          connectionstyle='arc3,rad=%s'%rad, 
          mutation_scale=10.0, 
          lw=2, 
          alpha=alpha, 
          color=color) 
     seen[(u,v)]=rad 
     ax.add_patch(e) 

    return e 


#======================================================== 

def animate(i): 
    startTime = time.time() 
    """perform animation step""" 
    global mainGraph, nodeToCoordinate_dict, label_coordinatePositions    

    mainGraph.add_node(str(i%20)+","+str(i/20), pos = (i%20,i/20)) 
    if not (i%20 == 0): 
     mainGraph.add_edge(str(i%20-1)+","+str(i/20), str(i%20)+","+str(i/20))   
     prevNode_coordinate = nodeToCoordinate_dict[str(i%20-1)+","+str(i/20)] 

     mainGraph[str(i%20-1)+","+str(i/20)][str(i%20)+","+str(i/20)]['p'] = 1.0 
    #END IF  
    nodeToCoordinate_dict[str(i%20)+","+str(i/20)] = (i%20,i/20)        
    ax=plt.gca() 
    draw_network(mainGraph,nodeToCoordinate_dict,ax) 
    ax.autoscale() 
    plt.axis('equal') 
    plt.axis('off')   
    print("time Elapsed = ", time.time()-startTime) 
    return None #mainGraph 

#==================================================== 

ani = animation.FuncAnimation(fig, animate, frames= 60,interval=1000) 
plt.show()

来源

2017-07-03 Sriram Gopalakrishnan

对不起 - 当我复制并粘贴这个，我得到一个图表显示单个节点，然后很多关键错误。 – Joel

@Joel这很奇怪，我将上面的代码复制到一个新的，孤立的脚本中，并按预期运行。你有需要的进口吗？也许更新的networkx和matplotlib。另外，请粘贴最相关的错误消息，我可能会弄清楚你的问题是什么 –

我想我想通了。使用“matplotlib.patches导入FancyArrowPatch，Circle”的函数正在减慢它的速度。这是“draw_network”功能。

我尝试用简单的代码（nx.draw（mainGraph，nodeToCoordinate_dict））替换绘图代码，并将时间间隔缩短为100（0.1秒）。像魅力一样工作

现在，我只需要想出一个加速曲线边缘图形的方法。

来源

2017-07-05 21:40:33

非常缓慢的网络图动画在Python的matplotlib与networkx

回答

相关问题