从生成器函数检索所有迭代器值

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