Python NetworkX 从作为根的节点在有向图中查找子图

我正在编写一个代码来从有向图中提取信息。该图也有循环。例如，

A->B->C->D
A->E->F->A
B->F->G

从这个图中，我想创建一个子图或节点列表，其中输入可以是任何节点，输出可以是输入节点是根的图，或者具有所有节点的节点列表输入节点的子节点（直到图的末尾）

例如，在上面的例子中， 1. 如果输入节点为C，则输出为D 2. 如果输入节点为B，则输出节点为C,D,F,G,A（由于存在环路，使得A到B是双向的） 3. 如果输入为 G，则输出为空白或 null。

python networkx 中有什么功能可以用来解决这个问题吗？

或者，有没有其他工具可以帮助我解决这个问题？

你想要的就是功能dfs_preorder_nodes() https://networkx.github.io/documentation/stable/reference/algorithms/generated/networkx.algorithms.traversal.depth_first_search.dfs_preorder_nodes.html#networkx.algorithms.traversal.depth_first_search.dfs_preorder_nodes。这是一个基于您的数据的小演示：

import networkx as nx

g = nx.DiGraph()

g.add_edge('A', 'B')
g.add_edge('B', 'C')
g.add_edge('C', 'D')

g.add_edge('A', 'E')
g.add_edge('E', 'F')
g.add_edge('F', 'A')

g.add_edge('B', 'F')
g.add_edge('F', 'G')

print('A:', list(nx.dfs_preorder_nodes(g, 'A')))
print('B:', list(nx.dfs_preorder_nodes(g, 'B')))
print('G:', list(nx.dfs_preorder_nodes(g, 'G')))

Output:

A: ['A', 'B', 'C', 'D', 'F', 'G', 'E']
B: ['B', 'C', 'D', 'F', 'A', 'E', 'G']
G: ['G']

输出包括起始节点。因此，如果您不想要它，只需从列表中删除第一个元素即可。

注意dfs_preorder_nodes()返回一个生成器对象。这就是为什么我打电话list()以获得可用的输出。