Как установить цвета для узлов в networkx python?

Я создал свой график, пока все выглядит великолепно, но я хочу обновить цвет своих узлов после создания.

Моя цель - визуализировать DFS, сначала я покажу начальный график, а затем поочередно раскрашиваю узлы, когда DFS решает проблему.

Благодаря

UPDATE

Если кому-то интересно, пример кода доступен на Github

Ответ 1

Все, что вам нужно, - это указать цветовую карту, которая отображает цвет для каждого node и отправить его функции nx.draw. Чтобы уточнить, для 20 node я хочу раскрасить первые 10 в синем, а остальные - зеленым. Код будет выглядеть следующим образом:

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

Вы найдете график в прикрепленном изображении .

Ответ 2

См. параметр node_color:

nx.draw_networkx_nodes(G, pos, node_size=200, node_color='#00b4d9')