2014-11-20 22 views
8

Grafiğimi oluşturdum, her şey çok güzel görünüyor, ancak oluşturulduktan sonra düğümlerimin rengini güncellemek istiyorum.Networkx python'daki düğümler için renkler nasıl ayarlanır?

Amacım DFS'yi görselleştirmek, önce DFS'nin sorunu çözdüğü sırada ilk grafiği ve ardından renk düğümlerini adım adım göstereceğim.

Teşekkür

GÜNCELLEME

kimse ilgileniyorsa, örnek kod Github

cevap

19

Tek ihtiyacınız her düğüme bir renk eşleştiren bir renk haritası belirlemek ve bunu göndermektir mevcuttur nx.draw işlevine. Netleştirmek gerekirse, 20 düğüm için ilk 10'u maviye ve kalanını yeşil renkte boyamak istiyorum.

G = nx.erdos_renyi_graph(20,0.1) 
color_map = [] 
for node in G: 
    if node <10: 
     color_map.append('blue') 
    else: color_map.append('green')  
nx.draw(G,node_color = color_map,with_labels = True) 
plt.show() 

ekteki resimdeki enter image description here grafiği bulacaksınız şöyledir: kod olacaktır.