為什么我的發電機 function 不能產生正確的值？

Question

我為我的編程 class 編寫了一個生成器 function 的任務，該生成器將一個 int ( n ) 和一個生成器 ( gen ) 作為 arguments 並生成給定生成器的前n元素。 我寫：

from typing import Iterator, Any

def basic_generator(n):  # Just a generator function to be able to test it out
    while True:
        yield n
        n += 1

def tryout(n: int, gen: Iterator[Any]) -> Iterator[Any]:
    i = 0
    while i < n:
        next(gen)
        yield i
        i = i + 1

在我收到的反饋中，它告訴我這是錯誤的，但沒有說明原因。 我知道它不起作用，但我不明白為什么。 有人可以向我解釋一下嗎？

Answer 1

您正在yield計數器而不是value ...更改為yield next(gen) 。

使用for循環而不是while也會更干凈：

def tryout(n: int, gen: Iterator[Any]) -> Iterator[Any]:
    for _ in range(n):
        yield next(gen)