Guest User

Untitled

a guest
Jan 18th, 2018
98
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.88 KB | None | 0 0
  1. import pandas as pd
  2. import numpy as np
  3. import networkx as nx
  4. import matplotlib.pyplot as plt
  5.  
  6. G = nx.Graph()
  7.  
  8. for ii in range(len(links_filtered)):
  9. a = data['var1'][ii]
  10. b = data['var2'][ii]
  11. c = data['value'][ii]
  12. G.add_edge(a,b,length=c,weight=c)
  13.  
  14. elarge = [(u,v) for (u,v,d) in G.edges(data=True) if d['weight']>0]
  15. esmall = [(u,v) for (u,v,d) in G.edges(data=True) if d['weight']<=0]
  16.  
  17. pos = nx.spring_layout(G,k=.2,iterations=10000)
  18.  
  19. nx.draw_networkx_nodes(G,pos,node_color='orange',node_size=400)
  20. nx.draw_networkx_edges(G,pos,edgelist=elarge,edge_color='blue')
  21. nx.draw_networkx_edges(G,pos,edgelist=esmall,edge_color='red',alpha=0.5,style='dashed')
  22. # nx.draw_networkx_labels(G,pos,font_size=8,)
  23.  
  24. for k,v in pos.iteritems():
  25. x,y = pos[k]
  26. plt.text(x,y,k,bbox=dict(facecolor='white',alpha=0.8),horizontalalignment='center',verticalalignment='baseline',fontsize=8,color='black')
Add Comment
Please, Sign In to add comment