Python中使用anytree库的NodeMixin()创建和操作具有层级关系的树状结构

anytree 是一个用于创建和操作层次关系树状结构的 Python 库。使用该库可以轻松地创建树状结构、查找节点、添加、删除和移动节点等操作。

首先，需要在 Python 中安装 anytree 库：

pip install anytree

接下来，可以通过继承 NodeMixin 类创建包含层次关系的树状结构。

以下是一个使用 anytree 库创建和操作树状结构的例子：

首先，导入需要的库：

from anytree import NodeMixin

然后，创建一个包含节点名称和父节点的自定义类。这个类需要继承 NodeMixin：

class MyNode(NodeMixin):
    def __init__(self, name, parent=None):
        super(MyNode, self).__init__()
        self.name = name
        self.parent = parent

现在可以使用自定义的类创建树状结构。以下是一个简单的树状结构示例：

# 创建树的根节点
root = MyNode("Root")

# 创建子节点
child1 = MyNode("Child 1", parent=root)
child2 = MyNode("Child 2", parent=root)
child3 = MyNode("Child 3", parent=root)

# 创建孙子节点
grandchild1 = MyNode("Grandchild 1", parent=child1)
grandchild2 = MyNode("Grandchild 2", parent=child1)

# 创建更深层级的节点
greatgrandchild1 = MyNode("Great-grandchild 1", parent=grandchild1)
greatgrandchild2 = MyNode("Great-grandchild 2", parent=grandchild1)

现在，我们已经创建了一个具有层级关系的树状结构，可以通过打印节点的名称来查看树的形状。

from anytree import RenderTree

print("整个树结构：")
for pre, fill, node in RenderTree(root):
    print("%s%s" % (pre, node.name))

输出：

整个树结构：
Root
├── Child 1
│   └── Grandchild 1
│       ├── Great-grandchild 1
│       └── Great-grandchild 2
├── Child 2
└── Child 3

然后，可以通过节点的名称来查找节点。

from anytree import search

node = search.find_by_attr(root, name="Child 1")
if node:
    print("找到节点: %s" % node.name)
else:
    print("未找到节点")

输出：

找到节点: Child 1

可以添加、删除或移动节点：

# 添加一个新的子节点
new_child = MyNode("New Child", parent=child1)

# 删除一个节点
child2.parent = None

# 移动节点
child3.parent = grandchild1

创建和操作树状结构就是这么简单！通过使用 anytree 库提供的功能，可以轻松地创建、操作和管理具有层级关系的树状结构。