我想为两个元素列表之间的每个可能组合创建一个列表,而不管列表的大小如何(它们的长度可以相同或可以不相同)。
我已经浏览了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
可以是任意大小。完全不关心格式,只需要完整的输出即可。
谢谢!
答案 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']