python - 如何确定函数返回值是否为生成器？

Question

yield 返回一个 collections.Iterable 但许多类型也是可迭代的

def aaaa():
    yield 1
    yield 2
    yield 3 

d = aaaa()

print(d,type(d),isinstance(d, collections.Iterable))

打印类型

<generator object aaaa at 0x0000000002626B88> <class 'generator'> True

我没有找到“生成器”类，或者我无法进行类型检查。

Answer 1

如果你想检查 iterable 是否是一个生成器对象:

import types
isinstance(aaaa(), types.GeneratorType) # ==> True

如果你想检查函数是否包含yield语句(即函数是一个生成器):

import inspect
inspect.isgeneratorfunction(aaaa) # ==> True