深入浅析Python中的迭代器


Posted in Python onJune 04, 2019

目录结构:

contents structure [-]

在开始文章之前,先贴上一张Iterable、Iterator与Generator之间的关系图:

深入浅析Python中的迭代器 

1. Iterator VS Iterable

迭代器(Iterator)

迭代器是实现了迭代器协议的类对象,迭代器协议规定了迭代器类必需定义__next()__方法。当对迭代器对象调用next()方法时,对象会去调用__next()__计算迭代器的返回值。

可迭代对象(Iterable)

可迭代对象可以是任何对象,不一定是能返回迭代器的数据结构。一个可迭代对象会直接或间接性的调用这两个方法__iter()__和__next()__;其中__iter()__方法只能返回迭代器对象,__next()__则供给迭代器进行调用。

通常情况下,可迭代类都会实现__iter()__和__next()__,并且__iter()__返回它自己,换句话说,该类即是迭代器又是可迭代类。

下面的代码展示了迭代器和可迭代器对象之间的差别:

a_set = {1, 2, 3}#定义set数据类型,set是可迭代类型
b_iterator = iter(a_set)#得到set的迭代器
#Output: 1
print(next(b_iterator))
#Output: <class 'set'>
print(type(a_set))
#Output: <class 'set_iterator'>
print(type(b_iterator))

从结果可以看出a_set是一个可迭代类型(set类型),b_iterator是一个迭代器(set_iterator),它们两个是完全不一同的类型。

下面的自定义了一个迭代器:

class Series(object):
 def __init__(self, low, high):
  self.current = low
  self.high = high
 def __iter__(self):
  return self
 def __next__(self):
  if self.current > self.high:
   raise StopIteration
  else:
   self.current += 1
   return self.current - 1
n_list = Series(1,10) 
print(list(n_list))

从上面的代码可以看出,__iter__返回了迭代器本身。__next__返回迭代器的下一个值,如果没有下一个返回值那么会抛出StopIteration异常。如果没有在合适的位置抛出StopIteration异常结束迭代,那么在某些循环语句中(例如:for loop),将会形成死循环,所以在__next__中必需要在合适位置添加退出语句(抛出StopIterator异常)。

2.Itertools 模块

Itertools是Python的内置模块,其中包含了能够创建迭代器的函数。简而言之,它提供了许多能够与迭代器交互的方法。

下面是我们使用Itertools模块中count函数的案例:

from itertools import count
sequence = count(start=0, step=1)
while(next(sequence) <= 10):
 print(next(sequence),end=" ")

输出:

Itertools中的cycle函数可以创建无限迭代器,例如:

from itertools import cycle
dessert = cycle(['Icecream','Cake'])
count = 0
while(count != 4):
 print('Q. What do we have for dessert? A: ' + next(dessert))
 count+=1

输出:

Q. What do we have for dessert? A: Icecream
Q. What do we have for dessert? A: Cake
Q. What do we have for dessert? A: Icecream
Q. What do we have for dessert? A: Cake

关于更多itertools模块的使用, 可以参见python文档 。

3.生成器(Generator)

生成器可以说是迭代器的亲兄弟,生成器允许我们像上面那样写迭代器而不用额外定义__iter__()和__next__()方法。

看下面的案例:

def series_generator(low, high):
 while low <= high:
  yield low
  low += 1
n_list = []
for num in series_generator(1,10):
 n_list.append(num)
print(n_list)

如果一个方法中出现了yield关键字,那么该方法就是一个生成器。生成器中没有return语句,函数的返回值实际上是一个generator。当循环开始执行到yield语句后,low的值会被扩展到要返回的generator中。当下一次循环到达yield语句时,generator会从上一次停止的地方恢复执行,并且将最新的low值添加到generator中。循环一直运行下去,直到low>high退出循环。

生成器支持延迟计算,只有当去取生成器中的值时才会计算。

例如:

def test():
 print("进入test函数")
 for i in range(2):
  print("yield number ",i)
  yield i
if "__main__" == __name__:
 print("开始调用test")
 res = test()
 print("结束调用test")
 next(res)
 next(res)

输出:

开始调用test
结束调用test
第一次next(res)
进入test函数
yield number 0
第二次next(res)
yield number  1

从结果可以看出,只有使用next调用迭代器时(使用for,while循环也可以),才会去执行迭代器函数中的内容。

python中生成器可以分为生成器函数和生成器表达式,生成器函数和生成器表达式是两种不同的类型。

生成器函数是一个函数体中有yield关键字的,我们上面定义的test就是生成器函数。

生成器表达式的使用比较受限制,一个生成器表达式返回一个生成器。下面是一个使用生成器表达式的案例:

squares = (x * x for x in range(1,10))
print(type(squares))
print(list(squares))

输出:

<class 'generator'>
[1, 4, 9, 16, 25, 36, 49, 64, 81]

生成器的效率是非常高的,生成器可以更好的利用内存和CPU的使用效率,并且通常生成器的代码都比较少,这使用生成器的代码非常好容易理解。应此应该尽量多的在代码中使用生成器

参考文档

https://www.datacamp.com/community/tutorials/python-iterator-tutorial

总结

以上所述是小编给大家介绍的Python中的迭代器,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对三水点靠木网站的支持!
如果你觉得本文对你有帮助,欢迎转载,烦请注明出处,谢谢!

Python 相关文章推荐
python之import机制详解
Jul 03 Python
Python用61行代码实现图片像素化的示例代码
Dec 10 Python
深入浅析Python中的迭代器
Jun 04 Python
python pyinstaller 加载ui路径方法
Jun 10 Python
python识别图像并提取文字的实现方法
Jun 28 Python
tf.concat中axis的含义与使用详解
Feb 07 Python
python实现堆排序的实例讲解
Feb 21 Python
Django数据库操作之save与update的使用
Apr 01 Python
利用python下载scihub成文献为PDF操作
Jul 09 Python
Python爬取12306车次信息代码详解
Aug 12 Python
详解python内置模块urllib
Sep 09 Python
python Matplotlib数据可视化(1):简单入门
Sep 30 Python
Python学习笔记之读取文件、OS模块、异常处理、with as语法示例
Jun 04 #Python
Python利用sqlacodegen自动生成ORM实体类示例
Jun 04 #Python
Python批量生成幻影坦克图片实例代码
Jun 04 #Python
python和mysql交互操作实例详解【基于pymysql库】
Jun 04 #Python
Python获取基金网站网页内容、使用BeautifulSoup库分析html操作示例
Jun 04 #Python
Python使用MyQR制作专属动态彩色二维码功能
Jun 04 #Python
Python流行ORM框架sqlalchemy安装与使用教程
Jun 04 #Python
You might like
Yii2创建表单(ActiveForm)方法详解
2016/07/23 PHP
PHP lcfirst()函数定义与用法
2019/03/08 PHP
jQuery使用手册之三 CSS操作
2007/03/24 Javascript
js 页面输出值
2008/11/30 Javascript
JavaScript操作XML 使用百度RSS作为新闻源示例
2012/02/17 Javascript
js 限制数字 js限制输入实现代码
2012/12/04 Javascript
Js 时间函数getYear()的使用问题探讨
2013/04/01 Javascript
jquery 日期控件datepicker属性详细解析
2013/11/08 Javascript
js setTimeout()函数介绍及应用以倒计时为例
2013/12/12 Javascript
js动态往表格的td中添加图片并注册事件
2014/06/12 Javascript
javascript新闻跑马灯实例代码
2020/07/29 Javascript
全面了解JavaScirpt 的垃圾(garbage collection)回收机制
2016/07/11 Javascript
ztree实现左边动态生成树右边为内容详情功能
2017/11/03 Javascript
element ui 对话框el-dialog关闭事件详解
2018/02/26 Javascript
jQuery+CSS实现的标签页效果示例【测试可用】
2018/08/14 jQuery
layui2.0使用table+laypage实现真分页
2019/07/27 Javascript
基于JS实现父组件的请求服务过程解析
2019/10/14 Javascript
JS实现简单tab选项卡切换
2019/10/25 Javascript
vue ajax 拦截原理与实现方法示例
2019/11/29 Javascript
Vue自定义表单内容检查rules实例
2020/10/30 Javascript
详解Vue.js3.0 组件是如何渲染为DOM的
2020/11/10 Javascript
[03:01]DOTA2英雄基础教程 露娜
2014/01/07 DOTA
[39:32]2014 DOTA2国际邀请赛中国区预选赛 TongFu VS DT 第二场
2014/05/23 DOTA
介绍Python中的fabs()方法的使用
2015/05/14 Python
Python GUI Tkinter简单实现个性签名设计
2018/06/19 Python
Python之list对应元素求和的方法
2018/06/28 Python
python如何导出微信公众号文章方法详解
2020/08/31 Python
IE支持HTML5的解决方法
2009/10/20 HTML / CSS
用html5实现语音搜索框的方法
2014/03/18 HTML / CSS
美国气象仪器、花园装饰和墙壁艺术商店:Wind & Weather
2019/05/29 全球购物
美国最大的户外装备和服装购物网站:Backcountry
2019/10/15 全球购物
聚网科技C++面试笔试题
2015/09/01 面试题
家长对孩子的感言
2014/03/10 职场文书
活动总结格式
2014/08/30 职场文书
公司车辆维修管理制度
2015/08/05 职场文书
小学四年级作文之人物作文
2019/11/06 职场文书