清单清单的所有组合


240

我基本上是在寻找组合的 python版本List<List<int>>

给定一个列表列表,我需要一个新列表,该列表给出列表之间所有可能的项目组合。

[[1,2,3],[4,5,6],[7,8,9,10]] -> [[1,4,7],[1,4,8],...,[3,6,10]]

列表的数量是未知的,因此我需要适用于所有情况的东西。优雅奖励积分!

Answers:


428

您需要itertools.product

>>> import itertools
>>> a = [[1,2,3],[4,5,6],[7,8,9,10]]
>>> list(itertools.product(*a))
[(1, 4, 7), (1, 4, 8), (1, 4, 9), (1, 4, 10), (1, 5, 7), (1, 5, 8), (1, 5, 9), (1, 5, 10), (1, 6, 7), (1, 6, 8), (1, 6, 9), (1, 6, 10), (2, 4, 7), (2, 4, 8), (2, 4, 9), (2, 4, 10), (2, 5, 7), (2, 5, 8), (2, 5, 9), (2, 5, 10), (2, 6, 7), (2, 6, 8), (2, 6, 9), (2, 6, 10), (3, 4, 7), (3, 4, 8), (3, 4, 9), (3, 4, 10), (3, 5, 7), (3, 5, 8), (3, 5, 9), (3, 5, 10), (3, 6, 7), (3, 6, 8), (3, 6, 9), (3, 6, 10)]

20
有人可以解释星号的含义*a吗?
塞拉诺

52
*a表示这些是传递给函数或方法的参数。def fn(a,b,c):将回应fn(*[1,2,3]) 参考
mjallday

1
@mjallday,是否还可以添加以下组合:(7,4,1),(8,4,1),(9,4,1),(10,4,1),(7,5, 1),(8,5,1),(9,5,1),(10,5,1)等?
重播

1
@Reman尚不清楚您想要获得什么,但是例如,如果它是每个元组的反向,则可以使用一个包装器函数,该函数以a输入作为输入,遍历itertools.product(*a)和处理yield由产生的元组itertools和反向版本(例如创建一个列表,reverse()然后将其转换回元组)。最好问一个新问题。
Joachim Wagner

24

最优雅的解决方案是在python 2.6中使用itertools.product

如果您不使用Python 2.6,itertools.product的文档实际上会显示等效的功能,以“手动”方式完成产品:

def product(*args, **kwds):
    # product('ABCD', 'xy') --> Ax Ay Bx By Cx Cy Dx Dy
    # product(range(2), repeat=3) --> 000 001 010 011 100 101 110 111
    pools = map(tuple, args) * kwds.get('repeat', 1)
    result = [[]]
    for pool in pools:
        result = [x+[y] for x in result for y in pool]
    for prod in result:
        yield tuple(prod)

19
listOLists = [[1,2,3],[4,5,6],[7,8,9,10]]
for list in itertools.product(*listOLists):
  print list;

希望您能像我初次接触时一样优雅。


5
那分号是怎么回事?:)
Paolo Bergantino,

3
习惯的力量。我喜欢Python如何让您使用分号,只是为了帮助我们的C / Java程序员。但是很明显; 当您执行诸如print(“ foo”);之类的操作时,它实际上不是语句终止符;这在C或Java中完全合法(尽管毫无意义),但在Python中被禁止。
马修·弗拉申

5

Numpy可以做到:

 >>> import numpy
 >>> a = [[1,2,3],[4,5,6],[7,8,9,10]]
 >>> [list(x) for x in numpy.array(numpy.meshgrid(*a)).T.reshape(-1,len(a))]
[[ 1, 4, 7], [1, 5, 7], [1, 6, 7], ....]

有人可以解释一下吗?
ashishv

5

直接执行此任务的递归没有什么问题,如果您需要使用字符串的版本,则可以满足您的需求:

combinations = []

def combine(terms, accum):
    last = (len(terms) == 1)
    n = len(terms[0])
    for i in range(n):
        item = accum + terms[0][i]
        if last:
            combinations.append(item)
        else:
            combine(terms[1:], item)


>>> a = [['ab','cd','ef'],['12','34','56']]
>>> combine(a, '')
>>> print(combinations)
['ab12', 'ab34', 'ab56', 'cd12', 'cd34', 'cd56', 'ef12', 'ef34', 'ef56']

3

一个人可以为此使用基本的python。该代码需要一个函数来拉平列表的列表:

def flatten(B):    # function needed for code below;
    A = []
    for i in B:
        if type(i) == list: A.extend(i)
        else: A.append(i)
    return A

然后可以运行:

L = [[1,2,3],[4,5,6],[7,8,9,10]]

outlist =[]; templist =[[]]
for sublist in L:
    outlist = templist; templist = [[]]
    for sitem in sublist:
        for oitem in outlist:
            newitem = [oitem]
            if newitem == [[]]: newitem = [sitem]
            else: newitem = [newitem[0], sitem]
            templist.append(flatten(newitem))

outlist = list(filter(lambda x: len(x)==len(L), templist))  # remove some partial lists that also creep in;
print(outlist)

输出:

[[1, 4, 7], [2, 4, 7], [3, 4, 7], 
[1, 5, 7], [2, 5, 7], [3, 5, 7], 
[1, 6, 7], [2, 6, 7], [3, 6, 7], 
[1, 4, 8], [2, 4, 8], [3, 4, 8], 
[1, 5, 8], [2, 5, 8], [3, 5, 8], 
[1, 6, 8], [2, 6, 8], [3, 6, 8], 
[1, 4, 9], [2, 4, 9], [3, 4, 9], 
[1, 5, 9], [2, 5, 9], [3, 5, 9], 
[1, 6, 9], [2, 6, 9], [3, 6, 9], 
[1, 4, 10], [2, 4, 10], [3, 4, 10], 
[1, 5, 10], [2, 5, 10], [3, 5, 10], 
[1, 6, 10], [2, 6, 10], [3, 6, 10]]

-1
from itertools import product 
list_vals = [['Brand Acronym:CBIQ', 'Brand Acronym :KMEFIC'],['Brand Country:DXB','Brand Country:BH']]
list(product(*list_vals))

输出:

[(''品牌缩写:CBIQ','品牌国家:DXB'),
('品牌缩写:CBIQ','品牌国家:BH'),
('品牌缩写:KMEFIC','品牌国家:DXB'),
( '品牌缩写:KMEFIC','品牌国家:BH')]


应该接受这个答案,因为它是唯一使用内置函数的答案,同时强调了它也适用于任何类型以及异构类型。
pedjjj

这个答案与许多年前有什么不同?
戴维德·拉斯祖克
By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.