弄懂这 5 个问题,拿下 Python 迭代器!

开发 后端
本篇专题写作思路,由几个问题,逐步讨论迭代器存在的价值、使用的方法、以及如何转化为自身武器并真正用到日常Python中。

本篇专题写作思路,由几个问题,逐步讨论迭代器存在的价值、使用的方法、以及如何转化为自身武器并真正用到日常Python中。

第一个问题:什么是迭代器?

迭代器,英文 Iterator,它首先是个对象,其次它是访问可迭代序列(Iterable)的一种方式。通常其从序列的第一个元素开始访问,直到所有的元素都被访问才结束。

迭代器又是一个特殊的对象,特殊在于它必须实现两个方法:__iter__和__next__.

第二个问题:为什么要有迭代器?

迭代器存在的一个最重要价值:节省内存,这在小数据量时无足轻重。

但是,当数据量大或者对程序要求性能高时,它的价值凸显。

第三个问题:迭代器怎么节省内存了?

只知道使用迭代器能节省内存,但却不知道怎么使用才能节省内存,下面就来回答这个问题。

首先创建一个list:

  1. In [1]: a=[1,3,5,9,10] 

其次,列表内每个元素+1,创建一个新列表

  1. In [2]: a1 = [i+1 for i in a]   

依次打印a1中每个元素:

  1. In [8]: for i in a1:  
  2.    ...:     print(i)  

上面操作等价于:

  1. a1 = [] 
  2. for i in a: 
  3.   a1.append(i+1) 
  4. for i in a1: 
  5.   print(i) 

空间复杂度为 O(n),n为列表a内元素个数。

但是,使用迭代器实现上面的元素+1并打印的空间复杂度是多少呢?

  1. ait = (i+1 for i in a) # 得到生成器,也是一种特殊的迭代器 
  2. for i in ait: 
  3.   print(i) 

上面操作等价于:

  1. for i in a: 
  2.   print(i+1) 

不需要额外空间,所以使用迭代器加1并打印的空间复杂度为O(1).

结论:迭代器更加节省空间!

第四个问题:如何自定义一个迭代器?

上面说过,迭代器对象必须要实现两个方法,为了更加具体,我们演示如何自定义一个迭代器。

自定义一个迭代器,实现斐波那契数列:

  1. #斐波那契数列 
  2. class Fabs(): 
  3.     def __init__(self,max): 
  4.         self.max=max 
  5.         self.n,self.a,self.b=0,0,1 
  6.      
  7.     #定义__iter__方法 
  8.     def __iter__(self): 
  9.         return self 
  10.          
  11.     #定义__next__方法 
  12.     def __next__(self): 
  13.         if self.n<self.max: 
  14.             tmp=self.b 
  15.             self.a,selfself.b=self.b,self.a+self.b 
  16.             self.n+=1 
  17.             return tmp 
  18.         raise StopIteration 

使用这个迭代器,打印斐波那契数列前10项:

  1. In [13]: for item in Fabs(10):  
  2.     ...:     print(item,end=' ')                                                                     
  3. 1 1 2 3 5 8 13 21 34 55  

第五个问题:迭代器使用有哪些注意事项?

迭代器只能前进不能回退!

也就是说一旦迭代结束,要想再使用此迭代器对象从头开始遍历元素,将是不可行的!

  1. In [14]: a=iter([1,4,5])  
  2.  
  3. In [16]: next(a)                                                                 
  4. Out[16]: 1 
  5.  
  6. In [17]: next(a)                                                                 
  7. Out[17]: 4 
  8.  
  9. In [18]: next(a)                                                                 
  10. Out[18]: 5 
  11.  
  12. # 要想再使用此迭代器对象从头开始遍历元素,将是不可行的! 
  13. In [19]: next(a)                                            
  14. # 抛出异常:StopIteration:  

使用内置函数iter,可将Iterable序列转化为迭代器。

最后总结:

  • 第一个问题:什么是迭代器?
  • 第二个问题:为什么要有迭代器?
  • 第三个问题:迭代器怎么节省内存了?
  • 第四个问题:如何自定义一个迭代器?
  • 第五个问题:迭代器使用有哪些注意事项?

 

责任编辑:赵宁宁 来源: Python与算法社区
相关推荐

2020-10-05 21:57:43

Python生成器可迭代对象

2016-03-28 10:39:05

Python迭代迭代器

2020-05-13 11:19:30

SaaS云计算技术

2021-01-10 23:36:52

SQL数据库技术

2017-06-26 16:26:15

Python迭代对象迭代器

2022-10-26 08:25:06

Python编程迭代器

2020-01-09 12:11:02

Python 开发编程语言

2021-11-28 08:03:41

Python迭代器对象

2018-08-30 07:03:49

2020-06-09 08:00:52

Python开发代码

2022-05-16 07:48:54

Python操作类型

2021-10-28 19:21:56

GolangGo变量

2020-07-26 00:40:48

JavaScript开发代码

2019-11-29 10:03:43

5G技术智能家居

2020-05-28 13:11:37

Python编码 开发

2021-01-14 13:37:27

Python数学统计

2020-07-16 08:04:21

浏览器缓存策略

2013-08-19 16:44:15

.Net

2022-10-26 09:27:59

Python编程迭代器协议

2015-11-10 12:24:36

创业问题思路
点赞
收藏

51CTO技术栈公众号