2013-07-13 8 views
5

Come si modifica il colore dei bordi in un grafico in retex in base ai pesi di quei bordi?Colorare i bordi networkx in base al peso

Il seguente codice fornisce solo tutti i bordi neri, anche se la mappa colori è il getto! (picture)

nx.draw_networkx(g,pos=pos,with_labels=True,edge_colors=[g[a][b]['weight'] for a,b in g.edges()], width=4,edge_cmap = plt.cm.jet) 

Scaling i pesi bordo per essere compreso tra 0 e 1 non cambia nulla.

io non sono sicuro di come il codice di cui sopra è diversa da quella in un related question solo che io non uso un ciclo per draw_networkx perché non sto animare il grafico.

risposta

2
#!/usr/bin/env python 
    """ 
    Draw a graph with matplotlib. 
    You must have matplotlib for this to work. 
    """ 
    try: 
     import matplotlib.pyplot as plt 
     import matplotlib.colors as colors 
     import matplotlib.cm as cmx 
     import numpy as np 
    except: 
     raise 

    import networkx as nx 

    G=nx.path_graph(8) 
    #Number of edges is 7 
    values = range(7) 
    # These values could be seen as dummy edge weights 

    jet = cm = plt.get_cmap('jet') 
    cNorm = colors.Normalize(vmin=0, vmax=values[-1]) 
    scalarMap = cmx.ScalarMappable(norm=cNorm, cmap=jet) 
    colorList = [] 

    for i in range(7): 
     colorVal = scalarMap.to_rgba(values[i]) 
     colorList.append(colorVal) 


    nx.draw(G,edge_color=colorList) 
    plt.savefig("simple_path.png") # save as png 
    plt.show() # display 

Appena modificato un codice di esempio da networkx che traccia un grafico semplice.