2010-07-17 8 views
7

Un produit de n copies d'un ensemble S est noté S n. Par exemple, {0, 1} est l'ensemble de toutes les séquences de 3 bits:Définir des produits en Python

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

Quel est le moyen le plus simple de répliquer cette idée en Python?

+0

est-ce juste {0,1} ou pourrait-il être quelque chose? – st0le

+0

Des ensembles arbitraires seraient bien, ainsi qu'un n arbitraire. – Eugene

Répondre

10

En Python 2.6 ou plus récent, vous pouvez utiliser itertools.product avec l'argument optionnel repeat:

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

Pour les anciennes versions de Python, vous pouvez mettre en œuvre product en utilisant le code trouvé dans la documentation:

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) 
1

Je suppose que cela fonctionne?

>>> 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)]) 
1

Mark, bonne idée.

>>> 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)]) 

Vous pouvez également étendre le type de jeu et de faire faire la méthode __pow__ cela.

0
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) 
Questions connexes