itertools and permutations - 合并两个列表以进行所有可能的组合

Question

我有两个列表： a和b 。 a是包含三个或更多字符串的列表，而b是分隔符列表。

我需要生成的，比合并与B的（参见更好地理解的例子）所有可能的组合结果中的所有可能的组合。

我希望列表a 中的所有三个项目都在结果的每一行中，以不同的顺序（为了覆盖或可能的组合）。 列表a 中的项目不能重复，而列表b 中的项目可以重复。

我不知道如何使它工作，所以你的帮助将不胜感激。

谢谢！

a = ["test1", "test2", "test3"]
b = ["_", "-", ".", ""]

# Expected result
test1_test2_test3
test2_test1_test3

. . . .

test1_test2-test3
test1_test2.test3

. . . .

test3test2_test1

. . . .

编辑：

from itertools import permutations, combinations, product, chain

a = ["test1", "test2", "test3"]
b = ["_", "-", ".", ""]

a_permutations = permutations(a)
b_permutations = product(b, repeat=len(a)-1)

for ap in a_permutations:
    for bp in b_permutations:
        result = ''.join([''.join(word) for word in zip(ap, bp)]) + ap[-1]
        print result

由于某种原因它不能正常工作..

Answer 1

由于您没有展示自己的代码，我只会给您一些想法。 如果您需要更多详细信息，请编辑您的问题以添加更多您自己的工作，然后询问详细信息。

当我查看您想要的列表时，您似乎希望列表a的 3 个项目不重复，以及列表b两个项目（可能重复），并将两个列表中的项目交错。 我还将假设列表b的项目数比给定列表a的长度少a 。 （如果这些假设不是您想要的，请在您的问题中明确说明。）

具有不同顺序而不重复的“组合”称为排列。 您可以使用表达式获得列表a所有排列

itertools.permutations(a)

因此列表b所有可能的项目集合与重复

itertools.product(b, repeat=len(a)-1)

对于每个集合中的每对项目，您可以通过对两个集合使用zip()操作来交错这些项目。 由于第一个集合比第二个集合长 1，因此最后一项将被排除在外，但您可以将其添加到结果中。

我可以给你看一个单线，可以进行这种交错，但我应该留下一些东西让你做。 使用这些想法来尝试您自己的代码。 如果您没有完全成功，请向我们展示一个尝试，我将很高兴向您展示一些最终代码。

请注意，遵循我的想法将导致列表的顺序与您显示的顺序不同。 我会得到的清单将从

test1_test2_test3
test1_test3_test2
test2_test1_test3
test2_test3_test1
test3_test1_test2
test3_test2_test1
test1_test2-test3
test1_test3-test2
test2_test1-test3
test2_test3-test1
test3_test1-test2
test3_test2-test1
test1_test2.test3

等等。 此顺序固定分隔符，然后排列列表a的项目。 当然，你可以反过来做，然后得到不同的顺序。

---

现在你已经展示了你自己的一些作品，这是我的代码。 这假设列表a和b没有重复。 这也有助于简化问题的处理，即比分隔符多一个项目——我只是将空分隔符""添加到分隔符元组的末尾。 然后两个长度（项目和分隔符）相等，每个字符串末尾的空分隔符什么也不做。

import itertools

a = ["test1", "test2", "test3"]
b = ["_", "-", ".", ""]

for separators in itertools.product(b, repeat=len(a) - 1):
    for items in itertools.permutations(a):
        print("".join(v for pair in zip(items, separators + ("",)) for v in pair))

此代码提供打印输出

test1_test2_test3
test1_test3_test2
test2_test1_test3
test2_test3_test1
test3_test1_test2
test3_test2_test1
test1_test2-test3
test1_test3-test2
test2_test1-test3
test2_test3-test1
test3_test1-test2
test3_test2-test1
test1_test2.test3
test1_test3.test2
test2_test1.test3
test2_test3.test1
test3_test1.test2
test3_test2.test1
test1_test2test3
test1_test3test2
test2_test1test3
test2_test3test1
test3_test1test2
test3_test2test1
test1-test2_test3
test1-test3_test2
test2-test1_test3
test2-test3_test1
test3-test1_test2
test3-test2_test1
test1-test2-test3
test1-test3-test2
test2-test1-test3
test2-test3-test1
test3-test1-test2
test3-test2-test1
test1-test2.test3
test1-test3.test2
test2-test1.test3
test2-test3.test1
test3-test1.test2
test3-test2.test1
test1-test2test3
test1-test3test2
test2-test1test3
test2-test3test1
test3-test1test2
test3-test2test1
test1.test2_test3
test1.test3_test2
test2.test1_test3
test2.test3_test1
test3.test1_test2
test3.test2_test1
test1.test2-test3
test1.test3-test2
test2.test1-test3
test2.test3-test1
test3.test1-test2
test3.test2-test1
test1.test2.test3
test1.test3.test2
test2.test1.test3
test2.test3.test1
test3.test1.test2
test3.test2.test1
test1.test2test3
test1.test3test2
test2.test1test3
test2.test3test1
test3.test1test2
test3.test2test1
test1test2_test3
test1test3_test2
test2test1_test3
test2test3_test1
test3test1_test2
test3test2_test1
test1test2-test3
test1test3-test2
test2test1-test3
test2test3-test1
test3test1-test2
test3test2-test1
test1test2.test3
test1test3.test2
test2test1.test3
test2test3.test1
test3test1.test2
test3test2.test1
test1test2test3
test1test3test2
test2test1test3
test2test3test1
test3test1test2
test3test2test1

Answer 2

这可以使用来自itertools permutations 、 product和combinations模块以pythonic方式实现，下面是工作片段：

from itertools import permutations, combinations, product

a = ["test1", "test2", "test3"]
b = ["_", "-", ".", ""]

for com in combinations(b, len(a) - 1):
    for per in product(com, repeat=len(a) - 1):
        for ear_per in permutations(a):
            out = ''.join(map(''.join, zip(list(ear_per[:-1]), per)))
            print(out + list(ear_per)[-1])