Python迭代器和生成器介绍
作者:junjie 时间:2022-03-22 07:41:24
迭代器
迭代器是一个实现了迭代器协议的对象,Python中的迭代器协议就是有next方法的对象会前进到下一结果,而在一系列结果的末尾是,则会引发StopIteration。
在for循环中,Python将自动调用工厂函数iter()获得迭代器,自动调用next()获取元素,还完成了检查StopIteration异常的工作。
常用的几个内建数据结构tuple、list、set、dict都支持迭代器,字符串也可以使用迭代操作。
你也可以自己实现一个迭代器,如上所述,只需要在类的__iter__方法中返回一个对象,这个对象拥有一个next()方法,这个方法能在恰当的时候抛出StopIteration异常即可。但是需要自己实现迭代器的时候不多,即使需要,使用生成器会更轻松。
#!/usr/bin/env python
# coding=utf-8
class test:
def __init__(self, input_list):
self.list = input_list
self.i = 0
def __iter__(self):
return self
def next(self):
if self.i == len(self.list):
self.i = 0
raise StopIteration
self.i += 1
return self.list[self.i - 1]
使用迭代器一个显而易见的好处就是:每次只从对象中读取一条数据,不会造成内存的过大开销。
例如:
/* 把文件一次加载到内存中,然后逐行打印。当文件很大时,这个方法的内存开销就很大了 */
for line in open("test.txt").readlines():
print line
/* 这是最简单也是运行速度最快的写法,他并没显式的读取文件,而是利用迭代器每次读取下一行 */
for line in open("test.txt"): #use file iterators
print line
生成器
生成器的编写方法和函数定义类似,只是在return的地方改为yield。
生成器中可以有多个yield。当生成器遇到一个yield时,会暂停运行生成器,返回yield后面的值。当再次调用生成器的时候,会从刚才暂停的地方继续运行,直到下一个yield。
生成器自身又构成一个迭代器,每次迭代时使用一个yield返回的值。
需要注意的是,生成器中不需要return语句,不需要指定返回值,在生成器中已经存在默认的返回语句
生成器表达式
(i for i in range(5))
// 返回迭代器
<generator object <genexpr> at 0x7ff3e8f0d960>
列表解析,返回list
[i for i in range(5)]
// 返回list
[0, 1, 2, 3, 4]
在这里存在一个问题,那就是range(5)会返回一个长度为5的数据,如果是range(1000)那么就会占用一个1000大小的数组空间;如果我们采用`生成器`,在需要的时候产生一个数字,那么空间的占用情况就会降低,这里我们可以使用xrange()函数来实现。
'''
xrange
函数说明:用法与range完全相同,所不同的是生成的不是一个数组,而是一个生成器。
xrange示例:
'''
>>> xrange(5)
xrange(5)
>>> list(xrange(5))
[0, 1, 2, 3, 4]
>>> xrange(1,5)
xrange(1, 5)
>>> list(xrange(1,5))
[1, 2, 3, 4]
>>> xrange(0,6,2)
xrange(0, 6, 2)
>>> list(xrange(0,6,2))
[0, 2, 4]
所以xrange做循环的性能比range好,尤其是返回很大的时候,尽量用xrange吧,除非你是要返回一个列表。
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
Javascript的一种模块模式
使用XSLT将XML数据转换成HTML
全面了解python中的类,对象,方法,属性
比较一下看看自己掌握了多少SQL快捷键
php初始化对象和析构函数的简单实例
高效地获取XMLhttp对象
详解Django中间件执行顺序
![](https://img.aspxhome.com/file/2023/0/85150_0s.png)
详解PHP中的mb_detect_encoding函数使用方法
Python 解析库json及jsonpath pickle的实现
![](https://img.aspxhome.com/file/2023/5/87925_0s.jpg)
response.setHeader()方法设置http文件头的值
python分别打包出32位和64位应用程序
![](https://img.aspxhome.com/file/2023/6/68296_0s.jpg)
该死的IE,走好
![](https://img.aspxhome.com/file/UploadPic/20091/15/byebye_ie6-71s.png)
利用Python实现简单的相似图片搜索的教程
![](https://img.aspxhome.com/file/2023/6/79466_0s.jpg)
详解Python字典的操作
![](https://img.aspxhome.com/file/2023/3/64643_0s.png)
C#数据导入/导出Excel文件及winForm导出Execl总结
php中数字、字符与对象判断函数用法实例
蜕变——记QQ医生3.0
![](https://img.aspxhome.com/file/UploadPic/20099/16/0909-qqdoctor3-02-88s.jpg)
Python容器类型公共方法总结
ASP 使用jqGrid实现读写删的代码(json)
python编程通过蒙特卡洛法计算定积分详解
![](https://img.aspxhome.com/file/2023/9/90809_0s.png)