如何为 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')