生成帶有networkx的有向圖的樹

Question

我在networkx中有一個有向圖G，我想得到它的最小生成樹。 我做：

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

這是無向的，我想恢復方向，但我不知道該怎么做。 我試過了：

G[T.edges()]

這最后一行看起來非常pythonic但這不是networkx的工作方式，顯然......有誰知道怎么做？

換句話說：如何給出（無向）邊緣的有向樹的子圖？

Answer 1

您可以通過簡單的理解獲得出現在MST T中的G邊緣：

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

然后，您可以從中構建新圖表。