Come impostare i colori per i nodi in python in retex?

Question

Ho creato il mio grafico, sembra tutto fantastico finora, ma voglio aggiornare il colore dei miei nodi dopo la creazione.

Il mio obiettivo è visualizzare DFS, prima mostrerò il grafico iniziale e poi i nodi di colore passo dopo passo mentre DFS risolve il problema.

Grazie

UPDATE

Se qualcuno è interessato, il codice di esempio è accessibile Github

Answer 1

Tutto ciò che serve è quello di indicare una mappa a colori che associa un colore ad ogni nodo e inviarlo alla funzione nx.draw. Per chiarire, per un nodo 20 voglio colorare i primi 10 in blu e il resto in verde. Il codice sarà il seguente:

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

troverete il grafico nell'immagine allegata .