NetworkX Draw Graphs

drawgraph.py

def drawgraph(graph, density=False):
  ## insert the graph you just converted from the edge list
  ## the default for the density is False. If density = True, it will return the density of the graph
  edges,weights = zip(*nx.get_edge_attributes(graph,'weight').items()) ## extract the edges and weights for later use
  ## k will affect the optimal distance between nodes.
  ## more types of layouts can be found at https://networkx.org/documentation/stable/reference/drawing.html
  pos=nx.spring_layout(graph,k=0.05,seed=42)
  nx.draw_networkx(graph,
                   pos,
                   with_labels=True,
                   node_size=400,
                   node_color="mistyrose",
                   edgelist=edges, 
                   edge_color=weights,
                   edge_cmap=plt.cm.Blues_r,
                   style="solid",
                   width=1)
  plt.subplots_adjust(left=2, bottom=3.2, right=6, top=6)

  if density:
    print("----------------------------------------")
    print("Density:",nx.classes.function.density(graph))
    print("----------------------------------------")

  return plt.show()