如何在python中返回2个列表的所有可能组合?

时间:2019-02-14 20:33:29

标签: python itertools

我想为两个元素列表之间的每个可能组合创建一个列表,而不管列表的大小如何(它们的长度可以相同或可以不相同)。

我已经浏览了itertools示例并搜索了stackoverflow,但是没有找到我要寻找的确切示例。

输入:

l1 = ['a', 'b', 'c']
l2 = [1, 2]

输出:

['a1-b1-c1', 'a1-b1-c2', 'a1-b2-c1', 'a1-b2-c2', 'a2-b1-c1', 'a2-b1-c2', 'a2-b2-c1', 'a2-b2-c2']

但同样,l1可以是任意大小,l2可以是任意大小。完全不关心格式,只需要完整的输出即可。

谢谢!

4 个答案:

答案 0 :(得分:2)

您可以使用/oauth生成{state}.edu/homepage的所有可能的3个组合,然后将每个组合与itertools.product合并。

l2

答案 1 :(得分:0)

您可以使用itertools.product并列出理解来获得您想要的结果:

from itertools import product
result = ["-".join(y+str(z) for y, z in zip(l1, x)) for x in product(*[l2] * 3)]

我想指出,这种方法不会产生任何额外的中间组合。

输出:

['a1-b1-c1', 'a1-b1-c2', 'a1-b2-c1', 'a1-b2-c2', 'a2-b1-c1', 'a2-b1-c2', 'a2-b2-c1', 'a2-b2-c2']

答案 2 :(得分:0)

您可以使用itertools.product

from itertools import product
['-'.join(map(''.join, zip(l1, c))) for c in product(map(str, l2), repeat=len(l1))]

这将返回:

['a1-b1-c1', 'a1-b1-c2', 'a1-b2-c1', 'a1-b2-c2', 'a2-b1-c1', 'a2-b1-c2', 'a2-b2-c1', 'a2-b2-c2']

答案 3 :(得分:0)

此解决方案从l1创建模板:例如'a{}-b{}-c{}'

from itertools import product
l1 = ['a', 'b', 'c']
l2 = [1, 2]
template = "{}-".join(l1) + "{}"
[template.format(*c) for c in product(l2, repeat=len(l1))]

['a1-b1-c1','a1-b1-c2','a1-b2-c1','a1-b2-c2','a2-b1-c1','a2-b1-c2 ','a2-b2-c1','a2-b2-c2']