深入浅析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 相关文章推荐
Djang中静态文件配置方法
Jul 30 Python
Python内置函数OCT详解
Nov 09 Python
Django进阶之CSRF的解决
Aug 01 Python
Python常见数据结构之栈与队列用法示例
Jan 14 Python
python使用socket实现的传输demo示例【基于TCP协议】
Sep 24 Python
Pyqt5自适应布局实例
Dec 13 Python
keras的ImageDataGenerator和flow()的用法说明
Jul 03 Python
使用python脚本自动生成K8S-YAML的方法示例
Jul 12 Python
python 常见的排序算法实现汇总
Aug 21 Python
Flask-SocketIO服务端安装及使用代码示例
Nov 26 Python
详解BeautifulSoup获取特定标签下内容的方法
Dec 07 Python
python 如何读、写、解析CSV文件
Mar 03 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
解决dede生成静态页和动态页转换的一些问题,及火车采集入库生成动态的办法
2007/03/29 PHP
PHP实现模仿socket请求返回页面的方法
2014/11/04 PHP
php中mkdir()函数的权限问题分析
2016/09/24 PHP
laravel中命名路由的使用方法
2017/02/24 PHP
Thinkphp5.0 框架视图view的比较标签用法分析
2019/10/12 PHP
PHP实现简单的计算器
2020/08/28 PHP
js 关键词高亮(根据ID/tag高亮关键字)案例介绍
2013/01/21 Javascript
JS获取键盘上任意按键的值(实例代码)
2013/11/12 Javascript
js实现用户注册协议倒计时的方法
2015/01/21 Javascript
详解JavaScript编程中的数组结构
2015/10/24 Javascript
浅析Javascript ES6中的原生Promise
2016/08/25 Javascript
BootStrap实现带有增删改查功能的表格(DEMO详解)
2016/10/26 Javascript
Bootstrap fileinput组件封装及使用详解
2017/03/10 Javascript
JointJS JavaScript流程图绘制框架解析
2019/08/15 Javascript
使用Vue-cli3.0创建的项目 如何发布npm包
2019/10/10 Javascript
vue中改变滚动条样式的方法
2020/03/03 Javascript
VUE : vue-cli中去掉路由中的井号#操作
2020/09/04 Javascript
Python实现抓取城市的PM2.5浓度和排名
2015/03/19 Python
Ruby元编程基础学习笔记整理
2016/07/02 Python
Python实现将多个空格换为一个空格.md的方法
2018/12/20 Python
python实现输入的数据在地图上生成热力图效果
2019/12/06 Python
基于keras 模型、结构、权重保存的实现
2020/01/24 Python
Python实现壁纸下载与轮换
2020/10/19 Python
Spartoo英国:欧洲最大的网上鞋店
2016/09/13 全球购物
美国在线健康和美容市场:Pharmapacks
2018/12/05 全球购物
FragranceNet中文网:北美健康美容线上零售商
2020/08/26 全球购物
什么是继承
2013/12/07 面试题
金融行业职业生涯规划范文
2014/01/17 职场文书
机修工工作职责
2014/02/21 职场文书
项目采购员岗位职责
2014/04/15 职场文书
目标责任书范本
2014/04/16 职场文书
监察建议书格式
2014/05/19 职场文书
给妈妈洗脚活动方案
2014/08/16 职场文书
迎国庆演讲稿
2014/09/15 职场文书
pytorch fine-tune 预训练的模型操作
2021/06/03 Python
MySQL示例讲解数据库约束以及表的设计
2022/06/16 MySQL