python - NetworkX中，如何获取包含节点数据的DFS树？

Question

在一个有向图中，节点不仅有 ID，还有数据(属性字典)，我想从给定的源节点开始获取 DFS 树，包括数据 的子节点。

但是，我注意到 nx.dfs_tree函数似乎返回一棵没有任何数据的树:

In [1]: import networkx as nx

In [2]: G = nx.DiGraph()

In [3]: G.add_nodes_from([(0, {'name': 'foo'}), (1, {'name': 'bar'}), (2, {'name
   ...: ': 'baz'})])

In [4]: G.add_edge(0, 1)

In [5]: G.add_edge(1, 2)

In [6]: T = nx.dfs_tree(G, 1)

In [7]: T.nodes[1]
Out[7]: {}

In [8]: T.nodes[2]
Out[8]: {}

In [9]: T.nodes
Out[9]: NodeView((1, 2))

In [10]: G.nodes[1]
Out[10]: {'name': 'bar'}

从上面的例子可以看出，T.nodes[1]是一个空字典，而G.nodes[1]包含的是最初传入的数据字典。

如何使 T.nodes[1] 和其他树节点包含与原始图相同的数据？

Answer 1

我最终使用 add_nodes_from 将数据从 G 复制到 T方法，更新现有属性:

In [31]: T.add_nodes_from((i, G.nodes[i]) for i in T.nodes)

In [32]: T.nodes[1]
Out[32]: {'name': 'bar'}

任何更优雅的解决方案将不胜感激。