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

您将在附图中找到图表 .