迭代器和生成器

一 迭代器

很多容器诸如列表、字符串可以用for循环遍历对象。for 语句会调用容器对象中的 iter()函数, 该函数返回一个定义了 next() 方法的迭代器对象,该方法将逐一访问容器中的元素。

python中,任意对象,只要定义了__next__方法,它就是一个迭代器。
python中的容器如列表、元组、字典、集合、字符串都可以被称作迭代器。
迭代就是从迭代器中取元素的过程。

列表是迭代器

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)

打开的text同样是迭代器

for line in open(“myfile.txt”):
print(line, end=’’)

如果你不想用for循环迭代呢?这时你可以:先调用容器(以字符串为例)的iter()函数再使用 next() 内置函数来调用 next() 方法当元素用尽时,next() 将引发 StopIteration 异常

s = "abds"
it = iter(s)
a = 1
if a == 1:
    print("迭代:"+next(it))
print(next(it))
print(next(it))
print(next(it))
print(next(it))

迭代器:a
b
d
s
StopIteration
结果只能输出了四个,然后报错。

二 生成器

廖雪峰:通过列表生成式,我们可以直接创建一个列表。但是,受到内存限制,列表容量肯定是有限的。
而且,创建一个包含100万个元素的列表,不仅占用很大的存储空间,如果我们仅仅需要访问前面几个元素,那后面绝大多数元素占用的空间都白白浪费了。
所以,如果列表元素可以按照某种算法推算出来,那我们是否可以在循环的过程中不断推算出后续的元素呢?
这样就不必创建完整的list,从而节省大量的空间。在Python中,这种一边循环一边计算的机制,称为生成器(Generator)。

生成器也是一种迭代器,但是你只能对其迭代一次。这是因为它们并没有把所有的值存在内存中,而是在运行时生成值。

生成器表达式生成器不一定要用复杂的函数表示,python提供了简洁的生成器表达式。从形式上来看,生成器表达式和列表推导式很像,仅仅是将列表推导式中的[]替换为(),但是两者差别挺大,生成器表达式可以说组合了迭代功能和列表解析功能。

生成器表达式可以认为是一种特殊的生成器函数,类似于lambda表达式和普通函数。但是和生成器一样,生成器表达式也是返回生成器generator对象,一次只返回一个值

>>> sum(i*i for i in range(10))                 # sum of squares
285

>>> xvec = [10, 20, 30]
>>> yvec = [7, 5, 3]
>>> sum(x*y for x,y in zip(xvec, yvec))         # dot product
260

转载:https://zhuanlan.zhihu.com/p/76831058

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值