生成列表列表的所有替换

Question

我正在 Python 中构建一个应用程序，我需要在其中定义以下类型的 function：

generate_replacements(['a', 'b', ['c', ['e', 'f']]], 1)

预期的 output 是输入列表的所有可能版本，其中仅替换了一个元素

[
[1, 'b', ['c', ['e', 'f']]],
['a', 1, ['c', ['e', 'f']]],
['a', 'b', 1],
['a', 'b', [1, ['e', 'f']]],
['a', 'b', ['c', 1]],
['a', 'b', ['c', [1, 'f']]],
['a', 'b', ['c', ['e', 1]]]
]

我可以看到递归是通往 go 的方式，但我真的很难弄清楚如何最好地开始这个。

Answer 1

您可以从列表中生成替换，然后如果您注意到您正在替换列表，也可以递归地将该列表传回 function。 如果您使用生成器，这会变得更简单：

def generate_replacements(l, rep):
    for i in range(len(l)):
        yield l[0:i] + [rep] + l[i+1: ]
        if isinstance(l[i], list):
            yield from (l[0:i] + [rec] + l[i+1: ] 
                        for rec in generate_replacements(l[i], rep))


list(generate_replacements(['a', 'b', ['c', ['e', 'f']]], 1))

这给：

[[1, 'b', ['c', ['e', 'f']]],
 ['a', 1, ['c', ['e', 'f']]],
 ['a', 'b', 1],
 ['a', 'b', [1, ['e', 'f']]],
 ['a', 'b', ['c', 1]],
 ['a', 'b', ['c', [1, 'f']]],
 ['a', 'b', ['c', ['e', 1]]]]