Récupération toutes les valeurs de iterator de la fonction générateur

Question

Disons que j'ai une fonction de générateur qui produit deux valeurs:

def gen_func(): 
    for i in range(5): 
     yield i, i**2 

Je veux récupérer toutes les valeurs de iterator de ma fonction. Actuellement, j'utilise cet extrait de code à cette fin:

x1, x2 = [], [] 
for a, b in gen_func(): 
    x1.append(a) 
    x2.append(b) 

Cela fonctionne pour moi, mais semble un peu maladroit. Existe-t-il un moyen plus compact de coder cela? Je pensais quelque chose comme:

x1, x2 = map(list, zip(*(a, b for a, b in gen_func()))) 

Ceci, cependant, me donne juste une erreur de syntaxe. PS: Je suis conscient que je ne devrais probablement pas utiliser un générateur à cette fin, mais j'en ai besoin ailleurs.

Modifier: N'importe quel type pour x1 et x2 fonctionnerait, cependant, je préfère la liste pour mon cas.

Answer 1

Si x1 et x2 peut être tuples, il est suffisant pour faire

>>> x1, x2 = zip(*gen_func()) 
>>> x1 
(0, 1, 2, 3, 4) 
>>> x2 
(0, 1, 4, 9, 16) 

Sinon, vous pouvez utiliser map pour appliquer list au iterator:

x1, x2 = map(list, zip(*gen_func())) 

Juste pour le plaisir, la même chose peut être fait en utilisant un déballage itérable prolongé:

>>> (*x1,), (*x2,) = zip(*gen_func()) 
>>> x1 
[0, 1, 2, 3, 4] 
>>> x2 
[0, 1, 4, 9, 16] 

Answer 2

Vous avait presque il:

x1, x2 = map(list, zip(*gen_func()))