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

Question

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

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

由于

更新

如果有兴趣，可以在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')