用生成器来改写直接返回列表的函数方法,生成器返回列表
用生成器来改写直接返回列表的函数方法,生成器返回列表
本文是一篇关于《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
需要注意的是:函数返回的迭代器,只能产生一轮结果,如果继续迭代第二轮,是不会再有结果的。
以上这篇用生成器来改写直接返回列表的函数方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持帮客之家。
相关内容
- python爬虫入门教程--快速理解HTTP协议(一),python爬虫
- Python操作使用MySQL数据库的实例代码,pythonmysql
- python爬虫入门教程--优雅的HTTP库requests(二),pythonr
- Python爬虫之模拟知乎登录的方法教程,python爬虫
- Python 含参构造函数实例详解,python构造函数
- Python 编码Basic Auth使用方法简单实例,pythonauth
- Python正则表达式完全指南,python完全指南
- python爬虫入门教程--利用requests构建知乎API(三),py
- python win32 简单操作方法,pythonwin32
- Python win32com 操作Exce的l简单方法(必看),win32comexce
评论关闭