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

クラス「ジェネレーター」が見つからないか、タイプチェックを実行できません。

score 10 · Accepted Answer

iterableがジェネレーターオブジェクトかどうかを確認したい場合:

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

関数にステートメントが含まれているかどうかを確認する場合yield(つまり、関数がジェネレーターである場合):

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

score 7 · Accepted Answer

inspect モジュール、特にinspect.isgeneratorfunctionを見てください。

そうは言っても、より興味深い質問は、それがリストなのか他のイテラブルなのかということです。その質問に答えるには、__getitem__が定義されているかどうかを確認するだけです。

2 に答える 2