Python 循环器(iterator)的功能

循环器(iterator)的功能

循环器（迭代器）是对象的容器，包含有多个对象。通过调用循环器的next()方法 (__next__()方法，在Python 3.x中)，循环器将依次返回一个对象。直到所有的对象遍历穷尽，循环器将举出StopIteration错误。

在for i in iterator结构中，循环器每次返回的对象将赋予给i，直到循环结束。使用iter()内置函数，我们可以将诸如表、字典等容器变为循环器。比如

for i in iter([2, 4, 5, 6]):

print(i)

 

>>> a=iter([2, 4, 5, 6])

>>> while True:

         next(a)

标准库中的itertools包提供了更加灵活的生成循环器的工具。这些工具的输入大都是已有的循环器。另一方面，这些工具完全可以自行使用Python实现，该包只是提供了一种比较标准、高效的实现方式。这也符合Python“只有且最好只有解决方案”的理念。

 

# import the tools

from itertools import *

无穷循环器

count(5, 2)     #从5开始的整数循环器，每次增加2，即5, 7, 9, 11, 13, 15 ...

cycle('abc')    #重复序列的元素，既a, b, c, a, b, c ...

repeat(1.2)     #重复1.2，构成无穷循环器，即1.2, 1.2, 1.2, ...

repeat也可以有一个次数限制:

repeat(10, 5)   #重复10，共重复5次

 

>>> a=cycle('abc')

>>> a

<itertools.cycle object at 0x0000000003141D48>

>>> next(a)

'a'

>>> next(a)

'b'

>>>

>>> next(a)

'c'