生成器、迭代器、可迭代对象

生成器

函数体中包含yield关键字的就是生成器
把生成器传给 next(...) 函数时，生成器函数会向前，执行函数定义体中的下一个 yield 语句，返回产出的值，并在函数定义体的当前位置暂停。等到再次遇到next才会继续执行。

简单生成器示例

def demo_gen():yield 1yield 2yield 3
gen = demo_gen()
print(next(gen))
print(next(gen))
print(next(gen))
print(next(gen))  # 当生成器的值取完了，在使用next方法，会抛出StopIteration错误

生成器表达式

生成器表达式是语法糖:完全可以替换成生成器函数，不过有时使用生成器表达式更便利。

代码在ipython(安装：pip install ipython)运行的

In [1]: (i for i in range(10))
Out[1]: <generator object <genexpr> at 0x107798200>

类示例代码：

class Sentence:"""简单的生成器，使用yield关键字"""def __init__(self, text):self.text = textself.words = RE_WORD.findall(self.text)def __iter__(self):for word in self.words:yield word

迭代器

使用`while`循环模拟迭代器

s = 'ABC'
it = iter(s)
while True:try:print(next(it))except StopIteration:del itbreak

实现迭代器的必备方法

__next__

返回下一个可用的元素，如果没有元素了，抛出StopIteration异常。
__iter__

返回self，以便在应该使用可迭代对象的地方使用迭代器，例如for循环中。

实现代码：

class IterableDemo:def __init__(self, text):self.text = textself.index = 0def __iter__(self):return iter(self)def __next__(self):"""next方法的获取逻辑"""try:word = self.text[self.index]except IndexError:raise StopIterationself.index += 1return wordif __name__ == '__main__':it = IterableDemo('Hello，world')print(next(it))print(next(it))print(next(it))

可迭代对象

可迭代对象是指可以通过__iter__和__getitem__方法访问。

其中__getitem__方法没有办法通过issubclass(Sentence, Iterable)判断。

解释器需要迭代对象x时，会自动调用iter(x)。

from collections.abc import Iterableclass Sentence:def __iter__(self):return iter(self)print(issubclass(Sentence, Iterable)) # == Trueclass Sentence:def __getitem__(self, index):return self.words[index]print(issubclass(Sentence, Iterable)) # == False

实现`getitem`方法

如果只实现了__getitem__方法，Python会创建一个迭代器，尝试按顺序（从索引0开始）获取元素。

如果尝试失败会返回C object is not iterable（C对象不可迭代），其中C对象就是所属的类。

def __getitem__(self, index):return self.words[index]

实现`iter`方法

def __iter__(self):return iter(self.words)

示例代码：

class Sentence:def __init__(self, text):self.text = textself.words = RE_WORD.findall(self.text)def __iter__(self):return iter(self.words)# def __getitem__(self, index):#     return self.words[index]def __len__(self):return len(self.words)def __str__(self):return 'Sentence(%s)' % reprlib.repr(self.words)s = Sentence('Hello world how are you')
for word in s:print(word)