使用 Python 创建递归拓扑树

Question

我有一个 df1 有两列代表链接：

point_1 point_2
  'A'     'B'
  'B'     'C'
  'C'     'D'
  'D'     'E'
  'D'     'F'
  'M'     'N'
  'N'     'O'
...

我还有另一个带有 point_A 和 end_point 的 df2（这是一个需要为该点中断三个的点​​），并且一些 point_A 可以有多个 end_point。

    point_A  end_point
      'A'        'E'
      'A'        'F'
      'M'        'O'
...

我不太确定如何实现它，我使用了一些函数和以下方式： 首先我将 df1 转换为列表：

temp = []
for row in df1.iterrows():
    index, data = row
    temp.append(data.tolist())

final_list = [[s.encode('ascii') for s in list] for list in temp]

现在的功能：

def all_paths(table, root):
    children = {}
    for node, child in table:
        if child: 
            children[node] = children.setdefault(node, []) + [child]

    def recurse(path):
        yield path
        if path[-1] in children:
            for child in children[path[-1]]:
                for foo in recurse(path + [child]):
                    yield foo

    return recurse([root])

path_list = []
for el in d:
    for i in el:
        for path in all_paths(final_list, i):
            path_list.append(path)

我得到了这个： 路径列表：

[['A'],
 ['A', 'B'],
 ['A', 'B', 'C'],
 ['A', 'B', 'C', 'D'],
 ['A', 'B', 'C', 'D', 'E'],
 ['A', 'B', 'C', 'D', 'F']
 ['M'],
 ['M', 'N'],
 ['M', 'N', 'O']
...
]

如您所见，我有很多列表项，但我不需要它们，我只需要从 point_A 到 end_point 的所有点的项。因此，我现在的想法是将列表中每个项目的第一个元素与 point_A 进行比较，并将列表中同一项目中的列表元素与 end_point 进行比较。如果相等，则表示它是正确的。

期望的输出：

[
 ['A', 'B', 'C', 'D', 'E'],
 ['A', 'B', 'C', 'D', 'F']
 ['M', 'N', 'O']
...
]

我会在数据框中做得更好，但这也可以。

但我确信有更简单、更正确的解决方案。 欢迎任何帮助。

Answer 1

point_A  end_point
  'A'        'E'
  'A'        'F'
  'M'        'O'

这个想法是你拿这张表并创建：

points = {'A':set(['E','F']), 'M':set(['O'])}

然后你循环这个：

for start_point, end_points in points.items():
    for path in all_paths(final_list, start_point, end_points):

然后在all_paths函数中：

替换

yield path

与：

if path[-1] in end_points:
   yield path

此测试会导致任何未在给定点之一结束的路径都不会被报告。

您可以通过查找链接的连接组件然后确保仅在同一组件中具有起点和终点的对的路径被枚举来加快这一速度。