web-dev-qa-db-ja.com

networkx pythonでノードの色を設定する方法は?

グラフを作成しましたが、これまでのところすべてが素晴らしく見えますが、作成後にノードの色を更新したいと思います。

私の目標は、DFSを視覚化することです。最初に最初のグラフを表示し、次にDFSが問題を解決するにつれてノードを段階的に表示します。

ありがとう

[〜#〜] update [〜#〜]

興味のある方は、サンプルコードを Github で入手できます。

24
Gokhan Arik

必要なのは、各ノードに色をマッピングして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()

添付の画像にグラフがあります enter image description here

46
Abdallah Sobehy