如何在networkx python中设置节点的颜色？

Question

我创建了我的图表,到目前为止一切看起来都很棒,但我希望在创建后更新节点的颜色.

我的目标是可视化DFS,我将首先显示初始图形,然后在DFS解决问题时逐步显示颜色节点.

谢谢

UPDATE

如果有人有兴趣,可以在Github上获得示例代码

Answer 1

您只需指定一个颜色映射,将颜色映射到每个节点并将其发送到nx.draw函数.为了澄清,对于一个20节点,我想要将前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()

您可以在附图中找到该图表.

Answer 2

参考node_color参数：

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