web-dev-qa-db-fra.com

Comment définir les couleurs des nœuds en python networkx?

J'ai créé mon graphique, tout semble bien jusqu'à présent, mais je veux mettre à jour la couleur de mes nœuds après la création.

Mon objectif est de visualiser DFS, je vais d'abord montrer le graphique initial puis colorer les nœuds étape par étape au fur et à mesure que DFS résout le problème.

Merci

[~ # ~] mise à jour [~ # ~]

Si quelqu'un est intéressé, un exemple de code est disponible sur Github

24
Gokhan Arik

Tout ce dont vous avez besoin est de spécifier une carte de couleurs qui mappe une couleur à chaque nœud et de l'envoyer à la fonction nx.draw. Pour clarifier, pour un nœud 20, je veux colorer les 10 premiers en bleu et les autres en vert. Le code sera le suivant:

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()

Vous trouverez le graphique dans l'image ci-jointe enter image description here .

46
Abdallah Sobehy