从发生器功能

Question

检索所有迭代器值比方说，我有这将产生两个值生成函数：

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

我想找回我的函数的所有迭代器值。目前我使用这个代码片段为此目的：

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

这对我有用，但似乎有点笨重。有没有更紧凑的编码方式？我在想：

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

但是，这只是给我一个语法错误。 PS：我知道我可能不应该使用一个生成器来达到这个目的，但我在其他地方需要它。

编辑：任何类型的x1和x2将工作，但是，我更喜欢列表我的情况。

Answer 1

如果x1和x2可元组，它足以做

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

否则，你可以使用map申请list迭代器：

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

只是为了好玩，同样的事情可以使用扩展迭代解包来完成：

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

Answer 2

你几乎已经：

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