如何为NetworkX中的节点设置颜色？

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