生成器函数,用于提供列表列表中的元素组合

时间:2018-11-22 11:18:21

标签: python combinations itertools

我有一个列表列表,我想找到所有组合,其大小为每个列表中每个元素可能出现的外部列表的长度。

例如:

input = [[1, 2], [3, 4], [5, 6]]
output = generateCombo(input)
#output should be 
[(1, 3, 5), (1, 3, 6), (1, 4, 5), (1, 4, 6), 
 (2, 3, 5), (2, 3, 6), (2, 4, 5), (2, 4, 6)] 

是否可以使用itertools来做到这一点?我尝试了 itertools.combinations ,但是它只能给我自己列表中的元素组合。

0 个答案:

没有答案