怎么理解Python迭代器与生成器？

怎么理解Python迭代器与生成器？在Python中，使用for ... in ... 可以对list、tuple、set和dict数据类型进行迭代，可以把所有数据都过滤出来。如下：

 

 

 

 

for element in [1, 2, 3]:

print(element)

for element in (1, 2, 3):

print(element)

for key in {'one':1, 'two':2}:

print(key)

for char in "123":

print(char)

for line in open("myfile.txt"):

print(line, end='')

 

1、迭代器

 

对字符串，列表，使用内置函数 iter 转化为可迭代对象， 使用next关键字，可迭代对象每次取其中的一个值。如下面代码：

 

nums = [1,2,3,4]

nums = iter(nums)

print(next(nums))

print(next(nums))

print(next(nums))

print(next(nums))

 

输出

 

1

2

3

4

 

如果超出数据的长度，将会报错 StopIteration

 

nums = [1,2,3,4]

nums = iter(nums)

print(next(nums))

print(next(nums))

print(next(nums))

print(next(nums))

print(next(nums)) # 超过数据长度，报错 StopIteration

 

2、生成器

 

使用生成器可以创建迭代器。生成器主要使用 yeild 关键字，每次调用 next 时会与yeild 定义的对应。例如：我们需要取 10～20之间的数据，步长为2，可以使用如下code

 

for x in range(10,20,2):

print(x)

但是如果步长设置为 0.5，如下代码：

for x in range(10,20,0.5):

print(x)

 

将会报错

 

TypeError: 'float' object cannot be interpreted as an integer

 

这个时候，我们就可以使用 生成器来自定义一个函数

 

def drange(start,stop,step):

x = start

while x< stop:

yield x

x += step

 

for x in drange(10,20,0.5):

print(x)

 

再举一个例子：

给定一个字符串，倒叙排列：

 

def reverse(data):

for x in range(len(data)-1,-1,-1):

yield data[x]

 

for x in reverse('gold'):

print(x)

 

看到这里，相信大家对于理解Python迭代器与生成器已经有了一定的理解。