python - 在 Python 中设置产品

Question

集合 S 的 n 个副本的乘积表示为 Sⁿ。例如，{0, 1}³ 是所有 3 位序列的集合:

{0,1}³ = {(0,0,0),(0,0,1),(0,1,0),(0,1,1), (1,0,0),(1,0,1),(1,1,0),(1,1,1)}

在 Python 中复制这个想法的最简单方法是什么？

Answer 1

在 Python 2.6 或更新版本中，您可以使用 itertools.product使用可选参数 repeat:

>>> from itertools import product
>>> s1 = set((0, 1))
>>> set(product(s1, repeat = 3))

对于旧版本的 Python，您可以使用文档中的代码实现 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)