问题描述
我正在尝试更改由 python 包 NetworkX 中的 dijkstras 算法访问的节点之间的边的颜色。这个项目中使用的grap是内置的nx.karate_club_graph()
如果我有它的代码,预期的结果是被访问节点之间的边的颜色将是红色。当前结果是所有访问过的节点都高亮显示,但边缘是默认的黑色。
我从堆栈溢出中查看了有关更改边缘颜色的不同解决方案,但我无法让它们适用于我的特定场景。
这是我制作的代码。
import networkx as nx
import matplotlib.pyplot as plt
G = nx.karate_club_graph()
nodes = G.nodes()
position = nx.spring_layout(G)
color = {"Mr. Hi": "#3DA3F5","Officer": "#E0D91B"}
dijkstra_route = nx.dijkstra_path(G,24,16)
colors = ["red" if n in dijkstra_route else color[G.nodes[n]["club"]] for n in nodes]
nx.draw(G,position,node_color = colors,with_labels = True)
解决方法
尝试将 edge_color
添加到 nx.draw()
import networkx as nx
import matplotlib.pyplot as plt
G = nx.karate_club_graph()
nodes = G.nodes()
position = nx.spring_layout(G)
color = {"Mr. Hi": "#3DA3F5","Officer": "#E0D91B"}
dijkstra_route = nx.dijkstra_path(G,24,16)
colors = ["red" if n in dijkstra_route else color[G.nodes[n]["club"]] for n in nodes]
dijkstra_edges = []
for index in range(0,len(dijkstra_route) - 1):
edge = (dijkstra_route[index],dijkstra_route[index + 1])
dijkstra_edges.append(edge)
edge_colors = ["red" if edge in dijkstra_edges else "black" for edge in G.edges()]
nx.draw(G,position,node_color=colors,edge_color=edge_colors,with_labels=True)
plt.show()
显示this,这与您想要的很接近,尽管我不明白为什么边缘(31,0)
没有着色