生成器和函数返回生成器之间的区别

2022-03-16 00:00:00 python generator

问题描述

我在使用生成器调试一些代码时遇到了这个问题。假设我有一个生成函数

def f(x):
    yield x

和返回生成器的函数:

def g(x):
    return f(x)

它们肯定返回相同的东西。在Python代码中互换使用它们会有什么不同吗?有没有办法区分这两者(没有inspect)?


解决方案

它们的行为相同。以及如何区分这两者(没有inspect)。用蟒蛇吗?仅检查:

import inspect

print inspect.isgeneratorfunction(g) --> False
print inspect.isgeneratorfunction(f) --> True

当然也可以使用dis

查看
Python 2.7.6 (default, Jun 22 2015, 17:58:13) 
[GCC 4.8.2] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> def f(x):
...     yield x
... 
>>> def g(x):
...     return f(x)
... 
>>> import dis
>>> dis.dis(f)
  2           0 LOAD_FAST                0 (x)
              3 YIELD_VALUE         
              4 POP_TOP             
              5 LOAD_CONST               0 (None)
              8 RETURN_VALUE        
>>> dis.dis(g)
  2           0 LOAD_GLOBAL              0 (f)
              3 LOAD_FAST                0 (x)
              6 CALL_FUNCTION            1
              9 RETURN_VALUE  

inspect更合适。

相关文章