通常接触到的迭代器有序列,比如字符串、列表等等,利用iter()方法可以构建一个迭代器,与之匹配的next()方法可以迭代返回迭代器内的值,并将返回值弹出迭代器。
In [1]: it = iter('python')
In [2]: print(it)
<str_iterator object at 0x00000187ADA75A00>
In [3]: next(it)
Out[3]: 'p'
In [4]: next(it)
Out[4]: 'y'
当next方法迭代到最后一个值时,继续调用next方法会弹出报错。当然也可以利用for循环迭代输出:
In [6]: for i in it:
...: print(i)
t
h
o
n
如果深入了解Python之后会发现生成器用起来会更加方便,结合函数即可。生成器的关键字为yield,生成器也具有next()方法,并且可以利用for循环迭代输出。
In [8]: def reverse(data):
...: for i in range(len(data)-1,-1,-1):
...: yield data[i]
In [9]: gen = reverse('mao')
In [10]: print(gen)
<generator object reverse at 0x00000187AD99FDD0>
In [11]: next(gen)
Out[11]: 'o'
In [12]: for i in gen:
...: print(i)
a
m
用小括号括起来的一个表达式可以用来创建一个生成器,下面将讲述如何来书写各种的表达式。
In [14]: gen2 = (i for i in range(0,5))
In [15]: print(gen2)
<generator object <genexpr> at 0x00000187ADA18D60>