如何使用 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-plet、5-plet 等？

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 Maynard 在评论中指出的那样，这不是笛卡尔积。为此，您必须检查 他的回答。