深入浅析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模拟鼠标拖动操作的方法
Mar 11 Python
python制作爬虫并将抓取结果保存到excel中
Apr 06 Python
使用Python向C语言的链接库传递数组、结构体、指针类型的数据
Jan 29 Python
python 同时读取多个文件的例子
Jul 16 Python
Django框架之登录后自定义跳转页面的实现方法
Jul 18 Python
python线程信号量semaphore使用解析
Nov 30 Python
python 爬虫 实现增量去重和定时爬取实例
Feb 28 Python
使用npy转image图像并保存的实例
Jul 01 Python
python如何删除列为空的行
Jul 17 Python
Python 发送邮件方法总结
Aug 10 Python
Python中glob库实现文件名的匹配
Jun 18 Python
基于PyQT5制作一个桌面摸鱼工具
Feb 15 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
PHP抓取及分析网页的方法详解
2016/04/26 PHP
thinkPHP内置字符串截取函数用法详解
2016/11/15 PHP
利用jQuery实现可输入搜索文字的下拉框
2013/10/23 Javascript
js实现交换运动效果的方法
2015/04/10 Javascript
JS实现同一个网页布局滑动门和TAB选项卡实例
2015/09/23 Javascript
JavaScript的ExtJS框架中数面板TreePanel的使用实例解析
2016/05/21 Javascript
Bootstrap和Java分页实例第二篇
2016/12/23 Javascript
基于vue实现分页/翻页组件paginator示例
2017/03/09 Javascript
javascript防篡改对象实例详解
2017/04/10 Javascript
Vue起步(无cli)的啊教程详解
2019/04/11 Javascript
vue实现手机号码的校验实例代码(防抖函数的应用场景)
2019/09/05 Javascript
Python中的字典遍历备忘
2015/01/17 Python
Python开发常用的一些开源Package分享
2015/02/14 Python
Python基于sftp及rsa密匙实现远程拷贝文件的方法
2016/09/21 Python
Python实现基于TCP UDP协议的IPv4 IPv6模式客户端和服务端功能示例
2018/03/22 Python
python 实现12bit灰度图像映射到8bit显示的方法
2019/07/08 Python
Python Web静态服务器非堵塞模式实现方法示例
2019/11/21 Python
实现Python与STM32通信方式
2019/12/18 Python
pytorch中的卷积和池化计算方式详解
2020/01/03 Python
python实现猜单词游戏
2020/05/22 Python
虚拟机下载python是否需要联网
2020/07/27 Python
python中实现栈的三种方法
2020/12/19 Python
传统HTML页面实现模块化加载的方法
2018/10/15 HTML / CSS
你可能不熟练的十个前端HTML5经典面试题
2018/07/03 HTML / CSS
HTML5操作WebSQL数据库的实例代码
2017/08/26 HTML / CSS
加拿大建筑和装修专家:Reno-Depot
2017/12/21 全球购物
给医务人员表扬信
2014/01/12 职场文书
2014年端午节活动方案
2014/03/11 职场文书
共产党员岗位承诺书
2014/05/29 职场文书
上班离岗检讨书
2014/09/10 职场文书
2015年社区纪检工作总结
2015/04/21 职场文书
党员带头倡议书
2015/04/29 职场文书
正规借条模板
2015/05/26 职场文书
仙境之桥观后感
2015/06/16 职场文书
react 项目中引入图片的几种方式
2021/06/02 Javascript
mysql中DCL常用的用户和权限控制
2022/03/31 MySQL