Establecer productos en Python

Question

Un producto de n copias de un conjunto S se denota S ⁿ. Por ejemplo, {0, 1} es el conjunto de todas las secuencias de 3 bits:

{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)}

¿Cuál es la forma más sencilla de replicar esta idea en Python?

Answer 1

En Python 2.6 o posterior puede utilizar itertools.product con el argumento opcional repeat:

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

Para versiones anteriores de Python se pueden implementar product utilizando el código que se encuentra en la documentación:

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) 

Answer 2

Supongo que esto funciona?

>>> s1 = set((0,1)) 
>>> set(itertools.product(s1,s1,s1)) 
set([(0, 1, 1), (1, 1, 0), (1, 0, 0), (0, 0, 1), (1, 0, 1), (0, 0, 0), (0, 1, 0), (1, 1, 1)]) 

Answer 3

Mark, buena idea.

>>> def set_product(the_set, n): 
    return set(itertools.product(the_set, repeat=n)) 

>>> s2 = set((0,1,2)) 
>>> set_product(s2, 3) 
set([(0, 1, 1), (0, 1, 2), (1, 0, 1), (0, 2, 1), (2, 2, 0), (0, 2, 0), (0, 2, 2), (1, 0, 0), (2, 0, 1), (1, 2, 0), (2, 0, 0), (1, 2, 1), (0, 0, 2), (2, 2, 2), (1, 2, 2), (2, 0, 2), (0, 0, 1), (0, 0, 0), (2, 1, 2), (1, 1, 1), (0, 1, 0), (1, 1, 0), (2, 1, 0), (2, 2, 1), (2, 1, 1), (1, 1, 2), (1, 0, 2)]) 

También puede extender el tipo de conjunto y hacer que el método __pow__ hacer esto.

Answer 4

print 'You can do like this with generator:' 
print set((a,b,c) for a in s1 for b in s1 for c in s1)