PandasPython的欧氏最小生成树

Question

我有一个数据帧，包括：

source  dest  euclidean
 A       B       0.5
 A       C       1.5
 A       D       0.5
 A       E       0.8
 B       C       0.5
 B       D       6.5
 B       E       5.4
 B       A       4.8
 C       B       4.3
 C       D       3.6
 C       E       2.6
 C       A       3.5
 D       B       8.0
 D       C       2.7
 D       E       7.7
 D       A       7.3

我想找到连接这些点的最小生成树，其中权重边是欧几里德距离。在

我尝试使用Geeks for geeks EMST中显示的方法：

^{pr2}$

但它给出了一个错误。在

我还有别的办法可以找到这个吗？任何线索都会有帮助的

Answer 1

使用networkx，可以使用

from networkx import *

g = Graph()
for index,row in df.iterrows():
  g.add_edge(row['source'],row['dest'],weight=row['euclidean'])

>>> list(minimum_spanning_edges(g))
[('A', 'E', {'weight': 0.8}),
 ('C', 'E', {'weight': 2.6}),
 ('C', 'D', {'weight': 2.7}),
 ('B', 'C', {'weight': 4.3})]