python - 在 Spring 布局+网络中更改节点的颜色

Question

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

所以我看到了这个link在 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)

现在，如果我有一个随机数并将其染成蓝色，让其他的染成红色......我将如何使用 'draw_networkx'< 的 nodes_list 属性来实现它 功能它有整数节点。如果您能为任何类型的节点名称提供解决方案，那将会更有帮助......

非常感谢..

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