如何使用python迭代器生成多個變量的笛卡爾積？

Question

親愛的所有人， 給定一個變量，例如三個值，我試圖生成這些變量的三元組的所有可能的組合。

儘管此代碼做的伎倆，

site_range=[0,1,2] 
states = [(s0,s1,s2) for s0 in site_range for s1 in site_range for s2 in site_range] 

它有點，嗯，笨拙，只在不斷惡化，如果我嘗試了三個多變量

的組合做同樣的。因此，我的Python 101問題：

1. 如何使用迭代器重寫上面的代碼？我的意思是，是否可以有一個迭代器來產生上述「狀態」的元素？

2. 是否有可能擴展這個不僅產生三胞胎，而且還有4人，5人等？

Answer 1

import itertools 
site_range=[0,1,2] 
[x for x in itertools.product(site_range, repeat=len(site_range))] 

Answer 2

使用itertools.product：

>>> site_range=[0,1] 
>>> list(product(site_range, repeat=3)) 
[000 001 010 011 100 101 110 111] 

編輯作爲@Glenn梅納德在評論中指出，這不是笛卡爾乘積。爲此，您必須檢查his answer。