How to set colors for nodes in networkx python?

Question

I created my graph, everything looks great so far, but I want to update color of my nodes after creation.

My goal is to visualize DFS, I will first show the initial graph and then color nodes step by step as DFS solves the problem.

Thanks

UPDATE

If anyone is interested, sample code is available on Github

Answer 1

All you need is to specify a color map which maps a color to each node and send it to nx.draw function. To clarify, for a 20 node I want to color the first 10 in blue and the rest in green. The code will be as follows:

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

You will find the graph in the attached image.