Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import pandas as pd
- import numpy as np
- import networkx as nx
- import matplotlib.pyplot as plt
- G = nx.Graph()
- for ii in range(len(links_filtered)):
- a = data['var1'][ii]
- b = data['var2'][ii]
- c = data['value'][ii]
- G.add_edge(a,b,length=c,weight=c)
- elarge = [(u,v) for (u,v,d) in G.edges(data=True) if d['weight']>0]
- esmall = [(u,v) for (u,v,d) in G.edges(data=True) if d['weight']<=0]
- pos = nx.spring_layout(G,k=.2,iterations=10000)
- nx.draw_networkx_nodes(G,pos,node_color='orange',node_size=400)
- nx.draw_networkx_edges(G,pos,edgelist=elarge,edge_color='blue')
- nx.draw_networkx_edges(G,pos,edgelist=esmall,edge_color='red',alpha=0.5,style='dashed')
- # nx.draw_networkx_labels(G,pos,font_size=8,)
- for k,v in pos.iteritems():
- x,y = pos[k]
- 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