将深度优先搜索转换为使用python进行面包优先搜索以进行决策树网络

Question

我已被指派实施决策树算法。到目前为止，我已经设法找到了所有节点/叶子，但我将它们放在一个平坦的1d阵列中。如何标记它们以便我可以提取视觉表示？ 我有最左边的部门首先搜索的节点列表，参考图片：[喜剧，枪支，叶子，律师，叶子，叶子，枪，叶子，叶子]

我需要的是像喜剧这样的标签：1支枪：1.1叶1.1.1律师：1.1.2等

我的代码基本上与此方案中的代码类似：

    track=0
    def create node(dataset):
       select best attribute
       save attribute with track #
       track+=1

       if this attribute has sub-options:
          track*=10 #to indicate new depth level
          create appropriate decision set
          create_node(newdataset)
       else:
          pass #interestingly this bottom part called multiple times

跟踪metohd没有帮助我，我认为它可以跟踪内部循环等但它失败了。任何建议都非常感谢