Python：从嵌套的字典和列表中递归提取评论树

Question

我正在尝试从由嵌套字典和列表组成的数据结构中提取一系列 cmets。贝娄是我正在处理的内容以及我到目前为止所写内容的一个示例：

example = {"text" : "Level 0", 
                "children" : [{"text": "Level 1",
                                "children" : [{"text": "Level 2a"}, 
                                              {"text": "Level 2b"}]}] }


def parse_comments(comment_tree): 

    if isinstance(comment_tree, list): 
        for tree in comment_tree: 
            return parse_comments(tree)
    if isinstance(comment_tree, dict): 
        if "children" in comment_tree.keys(): 
            return comment_tree["text"], parse_comments(comment_tree["children"])
        else: 
            return comment_tree["text"]

理想情况下，输出应如下所示：

[["Level 0", "Level 1", "Level 2a"], ["Level 0", "Level 1", "Level 2b"]]

我目前得到这个：

('Level 0', ('Level 1', 'Level 2a'))

有什么建议吗？

Answer 1

您可以将递归与生成器一起使用：

example = {"text" : "Level 0", "children" : [{"text": "Level 1","children" : [{"text": "Level 2a"}, {"text": "Level 2b"}]}] }
def get_comments(d, c = []):
  if 'children' not in d:
      yield c+[d['text']]
  else:
      for i in d['children']:
         yield from get_comments(i, c+[d['text']])

print(list(get_comments(example)))

输出：

[['Level 0', 'Level 1', 'Level 2a'], ['Level 0', 'Level 1', 'Level 2b']]