我重新发布这个问题是因为我被告知在上一篇文章中有一个解决方案。
我有 2 个列表:
list1 = ["foo", "bar", "lorem"]
list2 = ["X", "Y"]
我想从这两个列表中获得可能的组合,这意味着:
[["foo", "bar", "lorem"],
["foo", "bar", "loremX"],
["foo", "barX", "loremX"],
["fooX", "bar", "loremX"],
["fooX", "barX", "loremX"],
["foo", "barX", "lorem"],
["fooX", "barX", "lorem"],
["fooX", "bar", "lorem"],
["foo", "bar", "lorem"],
["foo", "bar", "loremY"],
["foo", "barY", "loremY"],
["fooY", "bar", "loremY"],
["fooY", "barY", "loremY"],
["foo", "barY", "lorem"],
["fooY", "barY", "lorem"],
["fooY", "bar", "lorem"]]
希望我没有错过任何组合。
有点迷失了这个。
它可能应该是 itertools.combinations_with_replacement
谢谢。
编辑
首先,感谢@titusarmah99 的出色回答。我设法采用了他的第二个非常简单的解决方案并使其通用:
import itertools
list1 = ["foo", "bar", "lorem"]
list2 = ["X", "Y"]
list2new = [""] + list2
newList = [[list1[i]+list2new[j] for j in range(len(list2new))] for i in range(len(list1))]
for index in range(1, len(list2) + 1):
for c in itertools.product([0,index],repeat=len(list1)):
tmp = [newList[i][c[i]] for i in range(len(c))]
print(tmp)