微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

如何为生成最小生成树的图的边着色

如何解决如何为生成最小生成树的图的边着色

我有一个包含 4 个节点的完整图 G。我需要为生成最小生成树的边缘着色。我怎样才能用 networkx 和 python 做到这一点?

解决方法

networkx.draw 接受一个可选的 edge_color 关键字参数,它允许您指定各个边的颜色。使用 minimum_spanning_tree 函数,我们可以将位于最小生成树内的边着色为红色,否则将其着色为黑色。

代码

import networkx as nx
G = nx.complete_graph(4)

mst = nx.minimum_spanning_tree(G)

edge_colors = ['red' if e in mst.edges else 'black' for e in G.edges]

nx.draw(G,edge_color=edge_colors)

输出

enter image description here

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。