Python:如何在仅更改特定元素时查找列表的所有组合

Question

我试图通过仅更改列表中的特定元素来使用 python 来查找列表的所有组合。例如，如果我有一个列表 [1,2,3,4,5,6,7,8,9] 并向其附加 3 个字符串，我会:

[1,2,3,4,5,6,7,8,9,'string','string','string']

然后我想找到当只允许更改字符串的位置时列表可以采用的所有组合。

例如

[1,2,3,4,5,'string',6,7,8,9,'string','string']
[1,2,'string',3,4,5,'string',6,7,'string',8,9]
['string',1,'string',2,3,4,'string',5,6,7,8,9]

等，同时仍将原始数字列表保持在相同的升序中。我不一定要一次存储所有简单的组合，我只是在尝试类似的操作:

• 遍历所有可能的组合
• 对于每种可能性，检查一个条件
• 如果为真，则将列表分配给一个变量
• 如果为false，则继续迭代

我一直在努力寻找一种解决方案，而不必使用不合理数量的 for 循环，并且适用于可能附加更多字符串的较大列表。我一直在考虑使用 itertools 但似乎找不到方法。

我找到的一个解决方案可能是只使用 itertools.permutations(带有附加字符串的列表)，然后使用条件来检查数字是否按升序排列，但我担心这种方法会效率真的很低，而且会占用大量内存，尤其是在处理较大的列表时。

如有任何帮助，我们将不胜感激，在此先致谢。

Answer 1

我想你可以在@wjandrea 的评论行中做一些事情:

from itertools import combinations, permutations

lst = [1,2,3,4,5,6,7,8,9]

strings = ['foo', 'bar', 'foobar']

for positions in combinations(range(len(lst) + len(strings)), len(strings)):
    for permutation in permutations(strings, len(strings)):
        cop = lst[:]
        for string, pos in zip(permutation, positions):
            cop.insert(pos, string)
        print(cop)

输出(小样本)

['foo', 'foobar', 1, 2, 3, 'bar', 4, 5, 6, 7, 8, 9]
['bar', 'foo', 1, 2, 3, 'foobar', 4, 5, 6, 7, 8, 9]
['bar', 'foobar', 1, 2, 3, 'foo', 4, 5, 6, 7, 8, 9]
['foobar', 'foo', 1, 2, 3, 'bar', 4, 5, 6, 7, 8, 9]
['foobar', 'bar', 1, 2, 3, 'foo', 4, 5, 6, 7, 8, 9]
['foo', 'bar', 1, 2, 3, 4, 'foobar', 5, 6, 7, 8, 9]
['foo', 'foobar', 1, 2, 3, 4, 'bar', 5, 6, 7, 8, 9]

请注意，此解决方案假定您可以重新排序 字符串。