Advertisement
Guest User

Untitled

a guest
Sep 18th, 2019
82
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.84 KB | None | 0 0
  1. def assign_colour(correlation):
  2. if correlation <= 0:
  3. return "#ffa09b" # red
  4. else:
  5. return "#9eccb7" # green
  6.  
  7.  
  8. def assign_thickness(correlation, benchmark_thickness=2, scaling_factor=3):
  9. return benchmark_thickness * abs(correlation)**scaling_factor
  10.  
  11.  
  12. def assign_node_size(degree, scaling_factor=50):
  13. return degree * scaling_factor
  14.  
  15.  
  16. # assign colours to edges depending on positive or negative correlation
  17. # assign edge thickness depending on magnitude of correlation
  18. edge_colours = []
  19. edge_width = []
  20. for key, value in nx.get_edge_attributes(Gx, 'correlation').items():
  21. edge_colours.append(assign_colour(value))
  22. edge_width.append(assign_thickness(value))
  23.  
  24. # assign node size depending on number of connections (degree)
  25. node_size = []
  26. for key, value in dict(Gx.degree).items():
  27. node_size.append(assign_node_size(value))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement