我需要一个 python 函数来深入搜索列表列表的字典答案

【问题标题】：I need a python function to depth first search into a dict of lists of lists我需要一个 python 函数来深入搜索列表列表的字典
【发布时间】：2021-06-14 18:02:04
【问题描述】：

我有这个数据结构，一个由列表组成的字典（一种树结构），其中的字母是短语：

dict = {
   'A': [ ['A1', ['A2'], ['A3', ['A4'], ['A5'] ] ],
   'B': [ ['B1', ['B2'] ],
   'C': [ ['C1', ['C2', ['C3'] ], ['C4'] ] ]
}

我需要的是一个函数，它返回“从根到叶的元素列表列表”。

换句话说，我需要沿着每条路径探索整个树，连接重建对话的短语。

这个树形结构是通过爬取 Reddit 构建的，因此每条路径都是一系列答案，因此是一个对话。

对于键“B”，有 1 个可能的序列：['B1', 'B2']（dict 键不相关）。

对于键 'A'，有 3 个可能的序列：['A1', 'A2'], ['A1', 'A3', 'A4'], ['A1', 'A3', 'A5' ]

对于键 'C'，有 2 个可能的序列：['C1', 'C2', 'C3'], ['C1', 'C4']

这个问题很难形式化，我希望这些序列是正确的:)

所以，我期望的是这个输出，一个包含树中所有可能序列的列表，以短语列表的形式：

[
   ['A1', 'A2'],
   ['A1', 'A3', 'A4'],
   ['A1', 'A3', 'A5'],
   ['B1', 'B2'],
   ['C1', 'C2', 'C3'],
   ['C1', 'C4']
]

我不需要结果列表中的任何特定顺序，我只需要序列。

非常感谢，每一个帮助都将不胜感激！

【问题讨论】：

欢迎来到 StackOverflow。展示您迄今为止尝试过的内容（代码），并具体说明哪个部分给您带来了麻烦。
您好，谢谢。我没有要发布的代码，因为我很难理解如何实现结果。我尝试了迭代和递归方法，但没有成功，因为我错过了如何做到这一点的主要理解，对不起。
Wrt " 我尝试了迭代和递归方法，但都没有成功" - 很好，将该代码添加到您的问题中。没关系，它没有工作或工作不好 - 否则你不会问这个问题:-)

标签： python list dictionary search depth

【解决方案1】：

这里是您问题的解决方案。

首先，我们创建一个函数，将您拥有的列表转换为字典结构：

def convert_to_dict(conversation_list):
    dict_ = {}
    for sublist in conversation_list:
        dict_[sublist[0]] = convert_to_dict(sublist[1:])
    return dict_

print(convert_to_dict([['A1', ['A2'], ['A3', ['A4'], ['A5']]]]))
# {'A1': {'A2': {}, 'A3': {'A4': {}, 'A5': {}}}}

然后我们创建一个函数来获取这个字典并返回其中的对话：

def conversations(rv, lst, conversation_dict):
    for key, value in conversation_dict.items():
        if value != {}:
            rv = conversations(rv, lst + [key], value)
        else:
            rv.append(lst + [key])
    return rv

print(conversations([], [], {'A1': {'A2': {}, 'A3': {'A4': {}, 'A5': {}}}}))
# [['A1', 'A2'], ['A1', 'A3', 'A4'], ['A1', 'A3', 'A5']]

然后我们可以将两者结合在一个函数中，该函数接受您的 dict 并返回所有对话：

def get_conversations(dict):
    rv = []
    for _, value in dict.items():
        dict_ = convert_to_dict(value)
        rv += conversations([], [], dict_)
    return rv

dict_ = {
   'A': [ ['A1', ['A2'], ['A3', ['A4'], ['A5'] ] ] ],
   'B': [ ['B1', ['B2'] ] ],
   'C': [ ['C1', ['C2', ['C3'] ], ['C4'] ] ]
}
print(get_conversations(dict_))
#[['A1', 'A2'], ['A1', 'A3', 'A4'], ['A1', 'A3', 'A5'], ['B1', 'B2'], ['C1', 'C2', 'C3'], ['C1', 'C4']]

【讨论】：

非常感谢！这正是我需要的，而且效果很好！你拯救了我的一天！

【解决方案2】：

您可以使用递归生成器函数来获得更短的解决方案：

d = {'A': [['A1', ['A2'], ['A3', ['A4'], ['A5']]]], 'B': [['B1', ['B2']]], 'C': [['C1', ['C2', ['C3']], ['C4']]]}
def get_paths(t):
    a, *b = t
    if not b:
       yield a
    yield from [[a, *(i if isinstance(i, list) else [i])] for j in b for i in get_paths(j)]

r = [i for [a] in d.values() for i in get_paths(a)]

输出：

[['A1', 'A2'], 
 ['A1', 'A3', 'A4'], 
 ['A1', 'A3', 'A5'], 
 ['B1', 'B2'], 
 ['C1', 'C2', 'C3'], 
 ['C1', 'C4']]

【讨论】：