用生成器来改写直接返回列表的函数

　　　本文是一篇关于《Effective Python》书中一节的学习笔记，记录了示例代码和思路。

　　如果函数要产生一系列结果，那么最简单的做法就是把这些结果都放在一个列表里返回。

　　比如我们要查出字符串中每个词的首字母在整串字符串中的位置：

def index_word(text):
    result=[]
    if text:
        result.append(0)
    for index,letter in enumerate(text):
        if letter == ' ':
            result.append(index+1)
    return result

　　该函数的使用：

　　

　　这个函数思路很明了，但存在的问题在于代码拥挤、冗余。返回前要把所有结果都放在列表里，如果输入量巨大，则程序可能会耗尽内存发生崩溃。

　　这个函数改用生成器（generator）来写会更好。可以对应任意长度的数据，不会影响到执行时所耗内存。

　　生成器是使用yield表达式的函数。调用生成器函数时，它并不会真的运行，而是会返回迭代器。每次在这个迭代器上面调用内置的next函数时，迭代器会把生成器推进到下一个yield表达式那里。生成器传给yield的每一个值，都会由迭代器返回给调用者。

def index_word_iter(text):
    if text:
        yield 0
    for index,letter in enumerate(text):
        if letter == ' ':
            yield index+1

　　

　　需要注意的是：函数返回的迭代器，只能产生一轮结果，如果继续迭代第二轮，是不会再有结果的。

转载于:https://www.cnblogs.com/lovealways/p/6899929.html