2

我目前正在使用anytree来生成我的搜索树，如何以列表格式从根节点开始获取所有可能的分支

from anytree import Node, RenderTree, AsciiStyle

f = Node("f")
b = Node("b", parent=f)
a = Node("a", parent=b)
d = Node("d", parent=b)
c = Node("c", parent=d)
e = Node("e", parent=d)
g = Node("g", parent=f)
i = Node("i", parent=g)
h = Node("h", parent=i)
print(RenderTree(f, style=AsciiStyle()).by_attr())

当前树：

f 
|-- b
|   |-- a
|   +-- d
|       |-- c
|       +-- e
+-- g
    +-- i
        +-- h

想要的输出（treeBranch）：

[[f,b,a], [f,b,d,c], [f,b,d,e], [f,g,i,h]]

我不确定是否有更好的方法可以做到这一点，欢迎提出任何建议。

我想使用这个列表来检查树中是否存在来自用户的新路径，例如：

newPath = [f, b]

for branch in treeBranch:
    if newPath in branch:
        return true
    else:
        // add new path to tree
4

3 回答 3

5

您希望拥有每个叶节点的根路径。只需使用PreOrderIter和 afilter_来检索叶节点：

print(list(PreOrderIter(f, filter_=lambda node: node.is_leaf)))
[a, c, e, h]

然后访问每个节点上的路径属性：

print([list(leaf.path) for leaf in PreOrderIter(f, filter_=lambda node: node.is_leaf)])
[[f,b,a], [f,b,d,c], [f,b,d,e], [f,g,i,h]]

如果您希望从树中的任何节点到叶节点的路径：

def allpaths(start):
    skip = len(start.path) - 1
    return [leaf.path[skip:] for leaf in PreOrderIter(start, filter_=lambda node: node.is_leaf)]
print(allpaths(b))
[(b, a), (b, d, c), (b, d, e)]

请注意，还有一个Walker，它提供从任何节点到另一个节点的路径。

于 2020-01-29T17:25:15.473 回答
0

看起来@c0fec0de 的解决方案在 anytree 2.8.0 中不再像以前那样工作了。我尝试了以下方法：

from anytree import Node, RenderTree, AsciiStyle, PreOrderIter
f = Node("f")
b = Node("b", parent=f)
a = Node("a", parent=b)
d = Node("d", parent=b)
c = Node("c", parent=d)
e = Node("e", parent=d)
g = Node("g", parent=f)
i = Node("i", parent=g)
h = Node("h", parent=i)
print(RenderTree(f, style=AsciiStyle()).by_attr())

这给了我以下树：

f
|-- b
|   |-- a
|   +-- d
|       |-- c
|       +-- e
+-- g
    +-- i
        +-- h

在此之后，要获取我尝试过的树中所有分支的列表：

>>> print(list(PreOrderIter(f, filter_=lambda node: node.is_leaf)))
[Node('/f/b/a'), Node('/f/b/d/c'), Node('/f/b/d/e'), Node('/f/g/i/h')]

由于响应位于节点对象中，要将其转换为字符串形式的节点列表，我执行了以下操作：

>>> lst = list(PreOrderIter(f, filter_=lambda node: node.is_leaf))
>>> print(lst)
[Node('/f/b/a'), Node('/f/b/d/c'), Node('/f/b/d/e'), Node('/f/g/i/h')]
>>> f = [str(i)[7:-2].split(i.separator) for i in lst]
>>> print(f)
[['f', 'b', 'a'], ['f', 'b', 'd', 'c'], ['f', 'b', 'd', 'e'], ['f', 'g', 'i', 'h']]

很想看看是否有更好的方法来做同样的事情。

于 2021-12-29T13:42:45.410 回答
0

@michael-westen 我完全利用 .name 和 .path 并避免字符串操作的解决方案：

leaves = [node_i.name for node_i in PreOrderIter(f.root, filter_=lambda node: node.is_leaf)]

branches = [[node_i.name for node_i in data_i.path] for data_i in PreOrderIter(f.root, filter_=lambda node: node.is_leaf)]

于 2022-01-17T12:17:07.357 回答