從生成器函數檢索所有迭代器值

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