《零基础入门学习Python》第048讲笔记：魔法方法：迭代器

TommyTimfy

支持迭代操作：字符串，字典，文件
iter()
next()

1. >>> it = iter("FishC")
   
2. >>> next(it)
   
3. 'F'
   
4. >>> next(it)
   
5. 'i'
   
6. >>> next(it)
   
7. 's'

复制代码

魔法方法

1. __iter__()
   
2. __next__() # 决定了迭代器的规则

复制代码

练习1

1. class Fibs:
   
2.     def __init__(self):
   
3.             self.a = 0
   
4.             self.b = 1
   
5.     def __iter__(self):
   
6.             # 本身就是迭代器，返回本身即可
   
7.         return self
   
8.     def __next__(self):
   
9.             self.a, self.b = self.b, self.a + self.b # 一一对应
   
10.             return self.a

复制代码

但是他是没有终止的，会不断地运行。
方法1：设置循环上限

1. fibs = Fibs()
   
2. for each in fibs:
   
3.     if each < 20:
   
4.         print(each)
   
5.     else:
   
6.         break

复制代码

方法2：参数设置

1. class Fibs:
   
2.     def __init__(self, n = 20):  # 可在这里简易调整
   
3.             self.a = 0
   
4.             self.b = 1
   
5.             self.n = n
   
6.     def __iter__(self):
   
7.             return self
   
8.     def __next__(self):
   
9.             self.a, self.b = self.b, self.a + self.b
   
10.             if self.a > self.n:
    
11.                 raise StopIteration
    
12.             return self.a

复制代码

注：若当前位置无元素，则抛出 StopIteration异常。
动动手：
0. 用while 实现以下内容

1. for each in range(5):
   
2.     print(each)

复制代码

1. alist = range(5)
   
2. it = iter(alist)
   
3. 
   
4. while True:
   
5.     try:
   
6.         print(next(it))
   
7.     except StopIteration:
   
8.         break

复制代码

1. 写一个迭代器，输出所有至今为止的闰年，效果如下

1. >>> leapYears = LeapYear()
   
2. >>> for i in leapYears:
   
3.         if i >=2000:
   
4.                 print(i)
   
5.         else:
   
6.                 break
   
7. 
   
8. 2012
   
9. 2008
   
10. 2004
    
11. 2000

复制代码

参考代码

1. import datetime as dt
   
2. 
   
3. class LeapYear:
   
4.         def __init__(self):
   
5.                 self.now = dt.date.today().year
   
6. 
   
7.         def isLeapYear(self, year):
   
8.                 if (year % 4 == 0 and year % 100 != 0) or (year % 400 == 0):
   
9.                         return True
   
10.                 else:
    
11.                         return False
    
12. 
    
13.         def __iter__(self):
    
14.                 return self
    
15. 
    
16.         def __next__(self):
    
17.                 # 不是闰年
    
18.                 while not self.isLeapYear(self.now):
    
19.                         self.now -= 1
    
20. 
    
21.                 temp = self.now
    
22.                 self.now -= 1
    
23. 
    
24.                 return temp

复制代码

2. 写一个MyRev类，功能 = reversed()，例如：

1. >>> myRev = MyRev("FishC")
   
2. >>> for i in myRev:
   
3.     print(i, end='')
   
4. 
   
5. ChsiF

复制代码

参考代码：

1. class MyRev:
   
2.         def __init__(self, data):
   
3.                 self.data = data
   
4.                 self.index = len(data)
   
5. 
   
6.         def __iter__(self):
   
7.                 return self
   
8. 
   
9.         def __next__(self):
   
10.                 if self.index == 0:
    
11.                         raise StopIteration
    
12. 
    
13.                 self.index -= 1
    
14.                 return self.data[self.index]
    
15. 
    
16. myRev = MyRev('FishC')
    
17. for i in myRev:
    
18.         print(i, end = '')

复制代码

jiajiaself

raise StopIteration 不应该会抛出一个异常吗？为什么运行的时候只是自动结束迭代，并没有抛出异常？