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中显示的方法：

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

g.KruskalMST()

但是它出错了。

我能找到其他任何方式吗？任何线索都会有所帮助

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})]

欧洲大陆最小生成树在熊猫python

1 个答案: