如何在python3中注释生成器？

Question

Python 3.x支持(可选)函数注释:

def add_ints(x:int, y:int) -> int :
    return x+y

我有时会遇到如何表示给定"类型"可以表示的问题,而这次,我有一个返回生成器的函数:

def myfunc(x: [int]) -> "generator that returns ints":
    #                     ^~~~~~~~~~~~~~~~~~~~~~~~~~
    return (n for n in x if n%2 == 0)

我该如何注释返回值？有什么参考我可以咨询吗？

Answer 1

虽然Generator[x, y, z]存在，但大多数情况下，您可能希望使用不那么冗长的Iterator：

def fn(x: int) -> Iterator[int]:
    return (n for n in range(x) if n%2 == 0)

也适用于 yield

def fn(x: int) -> Iterator[int]:
    for n in range(x):
        yield n

Answer 2

该类型模块定义的发电机类型,您可以使用这样的:

Generator[yield_type, send_type, return_type] 

另见PEP 0484.