创建具有多个子节点的Python OOP树(决策树)

class TreeNode:
    def __init__(self, value):
        self.value = value
        self.children = []

    def add_child(self, child):
        self.children.append(child)

class DecisionTree:
    def __init__(self, root):
        self.root = root

    def traverse(self, node):
        print(node.value)
        for child in node.children:
            self.traverse(child)

# 创建树节点
root_node = TreeNode("Root")
child1 = TreeNode("Child 1")
child2 = TreeNode("Child 2")
child3 = TreeNode("Child 3")

# 添加子节点
root_node.add_child(child1)
root_node.add_child(child2)
root_node.add_child(child3)

# 创建决策树
tree = DecisionTree(root_node)

# 遍历决策树
tree.traverse(tree.root)