在networkX中,我有一棵树作为DiGraph().
#!/usr/bin/python
# -*- coding: utf-8 -*-
import networkx as nx
t = nx.DiGraph()
t.add_edge(1,'r')
t.add_edge(2,'r')
t.add_edge(3,'r')
t.add_edge(4,2)
t.add_edge(5,2)
t.add_edge(6,5)
print t.edges()
如果采取树的节点2.
我怎么能得到2的子树?
编辑
我期待这个子树
[(4,2),(5,2),(6,5)]
解决方法:
如果你的意思是以节点2为根的子树,那就是
from networkx.algorithms.traversal.depth_first_search import dfs_tree
subtree_at_2 = dfs_tree(t, 2)
编辑:似乎你已经颠倒了边缘中节点的顺序.在有向树中,所有路径都从根到叶,而不是相反. dfs_tree(t.reverse(),2)为您提供了所需的树,但确实更改了您的代码.