[廖雪峰 - Python2.7教程] 返回函数2 - 学习笔记

在上一篇文章提到的例子中，还可以有这样的写法：

def count():

    for i in range(1, 4):
        def f():
            return i * i
        yield f

即通过生成器来迭代出count() 的数值，这里有一点需要注意，生成器的用法需要配合next() 或者for 语句迭代来使用，用法如下：

print count().next()

此时返回结果为：

<function f at 0x00000000026B95F8>

返回了yield 语句处的f 函数
也就是说可以这样用：

print (count().next())()

来输出第一个值

C:\Python27\python.exe C:/Python27/mystuff/0922_ClosePack.py
1

嗯，到这里结果看起来是对的。

连续输出呢？

print (count().next())()
print (count().next())()
print (count().next())()

C:\Python27\python.exe C:/Python27/mystuff/0922_ClosePack.py
1
1
1

结果是三个1...为什么会这样？根据结果来看，实际上只调用了一次生成器。也就是说count().next() 只调用了一次，可以这么理解，用括号括起来的count().next()实际上只指向了第一个next()，所以三次调用后都指向同一个空间，也即：

</pre><div><pre name="code" class="python">c = count().next()
print c()
print c()
print c()

两者结果是一样的：

C:\Python27\python.exe C:/Python27/mystuff/0922_ClosePack.py
1
1
1

这么理解起来就容易多了，c变量指向第一次调用的count().next()，然后三次调用其中保存的f 函数，这三次调用都没有用到新的next() 函数，所以结果都是1.

这样写也是错误的：

print count().next()()
print count().next()()
print count().next()()

正确的写法应该这样：

c = count()
print c.next()()
print c.next()()
print c.next()()

C:\Python27\python.exe C:/Python27/mystuff/0922_ClosePack.py
1
4
9

每次都会调用新的next() 函数。当然也可以用for 语句来迭代：

for f in count():
    print f()

推荐使用for 来迭代生成器，避免了很多不必要的语法错误。