迭代器

原理非常类似Lua的泛型for。

  1. for v in shit:
  2.     print v
  4. #for的幕后逻辑如下；
  5. iterable = iter(shit)    #生成对象的迭代器：iter()内置函数放回一个定义了__next__()方法的迭代对象
  6. v = next(iterable)        #进行迭代：next()内置函数会调用__next__方法返回迭代数据。
  7.                         #当迭代完了，next会抛出StopIteration的异常来终止for循环。
  9. #了解了for的幕后逻辑，我们可以自定义可迭代对象，而不仅限于使用内置类型。
  11. class IterableObj:
  12.     """文档说明字符串"""
  13.     def __init__(self, data):
  14.         self.data = data
  15.         self.index = len(data)
  17.     def __iter__(self):            #iter(obj)触发并返回值
  18.         return self
  20.     def __next__(self):            #next(obj)触发并返回值
  21.         if self.index == 0:
  22.             raise StopIteration
  23.         self.index = self.index - 1
  24.         return self.data[self.index]
  26. obj = IterableObj([1, 2, 3, 4, 5])
  27. for v in obj:
  28.     print(v)
  29. #5
  30. #4
  31. #3
  32. #2
  33. #1

生成器：yield

Generator，用于生成迭代器。写法类似函数，但当它们要返回数据时会使用 yield 语句，yield的效果就是每次next(生成器)时，就会从上次离开位置回复执行。
相比于上面手写迭代器，本质上就是自动完成了这些工作：

  • 自动创建方法
    • iter() 和 next() 方法。
  • 保存程序状态
    • 类似lua的闭包
  • 自动引发stopIteration

所以利用生成器，可以大大简化生成迭代器的工作。

  1. def reverse(data):
  2.     for index in range(len(data)-1, -1, -1):
  3.         yield data[index]
  5. for char in reverse('golf'):
  6.     print(char)

生成器表达式

类似列表推导式。

  1. sum(i*i for i in range(10))                #285
  4. xvec = [10, 20, 30]
  5. yvec = [7, 5, 3]
  6. sum(x*y for x,y in zip(xvec, yvec))     #260
  8. from math import pi, sin
  9. sine_table = {x: sin(x*pi/180) for x in range(0, 91)}
  11. unique_words = set(word for line in page  for word in line.split())
  13. valedictorian = max((student.gpa, student.name) for student in graduates)
  15. data = 'golf'
  16. list(data[i] for i in range(len(data)-1, -1, -1))    #['f', 'l', 'o', 'g']