python - 在春季布局+ networkx中更改节点的颜色

Question

如何更改位置未知的颜色节点

所以我在stackoverflow上看到了这个链接，其中给出了明确设置节点位置的答案。

但是说我有这样的图表

   G = nx.Graph()
   for i in range(10):
      G.add_node(i)
   for i in range(9):
      G.add_edge(i,i+1)
   pos = nx.spring_layout(G)
   node_num = random.randint(0,10)

但是，现在如果我有一些东西，例如获取一个随机数并将其涂成蓝色并让其他人为红色......我将如何使用函数nodes_list的属性来实现它'draw_networkx'这有整数节点。如果您可以为任何类型的节点名称提供解决方案，那将更有帮助....

多谢..

Answer 1

这是一些可能有帮助的代码。这会选择一个随机节点来着色为蓝色并将其余部分着色为红色。

import random
import matplotlib.pyplot as plt
import networkx as nx
G=nx.Graph()
G.add_edges_from([('a','b'),('b','d'),('c','e'),('b','e')])
pos=nx.spring_layout(G)
nodes=G.nodes()
random.shuffle(nodes)
blue=nodes.pop()
nx.draw_networkx_nodes(G,pos,nodelist=[blue],node_color='b')
nx.draw_networkx_nodes(G,pos,nodelist=nodes,node_color='r')
nx.draw_networkx_edges(G,pos)
plt.show()