不同尺寸的笛卡爾乘積

Question

我能有感謝名單的笛卡爾積到itertools.product（）函數：

lists = [['A', 'B'], ['1', '2'], ['x', 'y']] 
combinations = itertools.product(*lists) 
# [('A', '1', 'x'), ('A', '2', 'y'), ..., ('B', '2', 'y')] 

我要的是同樣的事情，但是各種大小：

all_comb = magicfunction(lists) 
# [('A', '1', 'x'), ..., ('B', '2', 'y'), ('A', '1'), ('A', '2'), ... ('2', 'y'), ... ('y')] 

我看不到一個明顯的方法來做到這一點。我需要一個方法，可以讓我設置元組的最小和最大尺寸（我處理長列表，只需要7到3的大小，列表數量和它們的大小的組合）。

我的名單更像是：

lists = [['A', 'B', 'C'], ['1', '2'], ['x', 'y', 'z', 'u'], ...] # size may go to a few dozens 

Answer 1

>>> from itertools import product, combinations 
>>> lists = [['A', 'B'], ['1', '2'], ['x', 'y']] 
>>> for i in xrange(2, len(lists)+1): 
    for c in combinations(lists, i): 
     print list(product(*c)) 
...   
[('A', '1'), ('A', '2'), ('B', '1'), ('B', '2')] 
[('A', 'x'), ('A', 'y'), ('B', 'x'), ('B', 'y')] 
[('1', 'x'), ('1', 'y'), ('2', 'x'), ('2', 'y')] 
[('A', '1', 'x'), ('A', '1', 'y'), ('A', '2', 'x'), ('A', '2', 'y'), ('B', '1', 'x'), ('B', '1', 'y'), ('B', '2', 'x'), ('B', '2', 'y')] 

Answer 2

只是連鎖幾個產品一起使用，基於更小尺寸的組合：

from itertools import chain, product, combinations 

def ranged_product(*lists, **start_stop): 
    start, stop = start_stop.get('start', len(lists)), start_stop.get('stop', 0) 
    return chain.from_iterable(product(*comb) 
           for size in xrange(start, stop - 1, -1) 
           for comb in combinations(lists, r=size)) 

演示：

>>> lists = [['A', 'B'], ['1', '2'], ['x', 'y']] 
>>> for prod in ranged_product(stop=2, *lists): 
...  print prod 
... 
('A', '1', 'x') 
('A', '1', 'y') 
('A', '2', 'x') 
('A', '2', 'y') 
('B', '1', 'x') 
('B', '1', 'y') 
('B', '2', 'x') 
('B', '2', 'y') 
('A', '1') 
('A', '2') 
('B', '1') 
('B', '2') 
('A', 'x') 
('A', 'y') 
('B', 'x') 
('B', 'y') 
('1', 'x') 
('1', 'y') 
('2', 'x') 
('2', 'y')