Iterator over lists comprehensions ?

Salut,

Je débute en python, et pour apprendre, je veux faire quelque chose de très simple : générer toutes les combinaisons de plusieurs ensembles (numériques).

Par exemple, je veux générer tous les vecteurs possibles, avec comme composante sur la 1ère dimension tout entier naturel inférieur à 2, et sur la 2e dimension tout entier naturel inférieur à 3.

Voici ce que j'ai codé :

Code:

---

1 2 3 4 5 6

def vectors(dimensions, index=0): if index == len(dimensions): return [[]] return [[h] + t for h in range(dimensions[index]) for t in vectors(dimensions, index + 1)]   print vectors([2,3])

---

Code:

---

1 2	./test.py [[0, 0], [0, 1], [0, 2], [1, 0], [1, 1], [1, 2]]

---

Ça marche avec n'importe quel nombre de dimensions, par exemple avec [2,3,2,2] :

Code:

---

1 2

./test.py [[0, 0, 0, 0], [0, 0, 0, 1], [0, 0, 1, 0], [0, 0, 1, 1], [0, 1, 0, 0], [0, 1, 0, 1], [0, 1, 1, 0], [0, 1, 1, 1], [0, 2, 0, 0], [0, 2, 0, 1], [0, 2, 1, 0], [0, 2, 1, 1], [1, 0, 0, 0], [1, 0, 0, 1], [1, 0, 1, 0], [1, 0, 1, 1], [1, 1, 0, 0], [1, 1, 0, 1], [1, 1, 1, 0], [1, 1, 1, 1], [1, 2, 0, 0], [1, 2, 0, 1], [1, 2, 1, 0], [1, 2, 1, 1]]

---

Maintenant, je voudrais savoir si ce code rendu très court grâce aux "lists comprehensions" peut être transformé en iterator, c'est à dire sans générer la liste complète, mais seulement à la demande ?

Est-il possible de combiner les deux (iterator + lists comprehensions) ?

PS: Pour info, j'ai réussi à coder le même résultat en iterator, mais sans utiliser les "lists comprehensions" :

Code:

---

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20

class CursorGenerator:   def __init__(self, dimensions): self.dimensions = dimensions   def cursor_increment(self, cursor, index=-1): if cursor[index] >= self.dimensions[index] - 1: if index == - len(self.dimensions): return False cursor[index] = 0 return self.cursor_increment(cursor, index - 1) else: cursor[index] += 1 return True   def __iter__(self): cursor = [0] * len(self.dimensions) yield cursor while self.cursor_increment(cursor): yield cursor

---

Merci de votre aide.

J'ai trouvé, grâce à la réactivité du chan IRC #python-fr :-)

Il suffit de mettre des parenthèses au lieu des crochets.

Code:

---

[i for i in range(10)]

---

est une liste.

Code:

---

(i for i in range(10))

---

est un générateur.

Ainsi, voici le code modifié :

Code:

---

1 2 3 4 5 6 7

def vectors(dimensions, index=0): if index == len(dimensions): return [[]] return ([h] + t for h in xrange(dimensions[index]) for t in vectors(dimensions, index + 1))   for v in vectors([2,3,2,2]): print v

---

Bonsoir,

Juste pour info, ton code utilise ce qu'on appelle une generator expression, mais on peut aussi coder explicitement un générateur avec le mot-clé yield, qui permet d'implémenter une forme de coroutine:

Code:

---

1 2 3 4 5 6 7

def vectors(dimensions, index=0): if index == len(dimensions): yield [] else: for h in xrange(dimensions[index]): for t in vectors(dimensions, index+1): yield [h] + t

---

L'exécution du générateur vectors retourne un itérateur; chaque appel de next sur cet itérateur exécute le code jusqu'au prochain yield, retourne la valeur indiquée, et le contexte d'exécution est sauvé pour être restauré au prochain appel de next.

Oui, c'est d'ailleurs bien puissant les yield expressions.

J'ai découvert et utilisé tout ça sur un solveur de cube-serpent : http://blog.rom1v.com/2011/09/resoud...ent-en-python/