zoukankan      html  css  js  c++  java
  • 用生成器来改写直接返回列表的函数

       本文是一篇关于《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

      

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

  • 相关阅读:
    编程是点滴的积累
    Tech.ED 2005 北京 第二天印象
    Tech.ED 2005 北京 第三天印象
    别把事情弄的太复杂
    在看《青衣》
    可以用的开源包
    KVM虚拟机的性能问题
    [zz]kvmlibvirt的使用:创建虚拟机与快照
    KVM快照snapshot
    [zz]kvm环境快照(snapshot)的使用方法
  • 原文地址:https://www.cnblogs.com/lovealways/p/6899929.html
Copyright © 2011-2022 走看看