python - 带networkx的有向图生成树

Question

我在 networkx 中有一个有向图 G，我想得到它的最小生成树。我这样做:

 T = nx.algorithms.minimum_spanning_tree( G.to_undirected()  )

这是无方向的，我想恢复方向，但我不知道该怎么做。我试过:

G[T.edges()]

这最后一行看起来很 pythonic 但这不是 networkx 的工作方式，显然......有谁知道怎么做吗？

换句话说:如何在给定(无向)边的情况下获得有向树的子图？

Answer 1

你可以通过简单的理解得到G中出现在MST T中的边:

E = set(T.edges())  # optimization
[e for e in G.edges() if e in E or reversed(e) in E]

然后您可以从中构建一个新图表。