我有一个输入字典-dict_input,目标为键,源为值.一个目的地可以有一个或多个来源.
dict_input = {'C411':['C052'],'C052':['C001','C002'], 'C001':['9001'], 'C002':['9002']}
在上面的dict_input中,终端目的地为C411,而初始来源为9001和9002.我试图为终端目的地C411提出来源路径.预期输出形式为列表-
[['C411', 'C052', 'C001', '9001'], ['C411', 'C052','C002', '9002']]
我有以下代码:
def get_source(node, dict_input, source=[]):
if node in dict_input:
source.append(node)
for i in dict_input[node]:
if i != node:
get_source(i, dict_input, source)
else:
source.append(node)
else:
source.append(node)
return source
return source
dict_input = {'C052':['C001','C002'], 'C411':['C052'], 'C001':['9001'], 'C002':['9002']}
print(get_source('C411', dict_input, []))
输出是将两个源路径组合到一个列表中-
['C411', 'C052', 'C001', '9001', 'C002', '9002']
如何修改代码以获取每个源路径的单独列表?