python 迭代器为什么无法第二次打印元素?

问题: python脚本中,迭代器为什么无法第二次打印元素?
先来看一个小程序:

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

#将字符串'12'转成数字12
dic = {'1':1,'2':2}
a = map(lambda ch:dic[ch],'12')
print(a)
#print(list(a))

print(a)
for x in a:
  print(x)

运行结果是:
这里写图片描述

以上运行结果,说明for循环生效了,做一个改动:for循环之前,print(list(a))
代码:

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

dic = {'1':1,'2':2}
a = map(lambda ch:dic[ch],'12')
print(a)
print(list(a))

print(a)
for x in a:
  print(x)

运行结果:
这里写图片描述

可以看到,for循环没有打印出a中的元素。why?是因为将a变为list导致的吗?直接将a遍历两次的结果会如何呢?

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

dic = {'1':1,'2':2}
a = map(lambda ch:dic[ch],'12')
print(a)
for x in a:
  print(x)

print(a)
print('the second time:')
for x in a:
  print(x)

运行结果:
这里写图片描述

第二次for循环没有打印a中的元素。。

**问题分析:** a是一个map object,print(list(a))或者for循环打印a中的元素,都会导致[迭代器](http://www.liaoxuefeng.com/wiki/0014316089557264a6b348958f449949df42a6d3a2e542c000/00143178254193589df9c612d2449618ea460e7a672a366000)从头走到尾(可以类比list[0]到list[n])。 而迭代器是一个[单向](http://stackoverflow.com/questions/17416777/why-can-i-only-use-a-reader-object-once)的容器,走到尾部之后,不会自动再回到开始位置。 所以,以上小程序中的print(list(a))或者对a进行一次for循环之后,再一次for循环不会打印出任何结果。 **解决办法:** 方法1:遍历map之后,将元素位置恢复到位置0. 可以参考[why-can-i-only-use-a-reader-object-once](http://stackoverflow.com/questions/17416777/why-can-i-only-use-a-reader-object-once) 方法2:复制两个map对象(使用itertools.tee()方法来复制iterator),两次循环时,各用一个 代码:

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import itertools

dict = {'1':1, '2':2}
a,b = itertools.tee(map(lambda ch:dict[ch],'12'))
print(a)
print(b)

for x in a:
    print(x)

for y in b:
    print(y)

运行结果:
这里写图片描述

可以看到,两个for循环都有打印元素。问题解决:)


参考:
http://stackoverflow.com/questions/17416777/why-can-i-only-use-a-reader-object-once

http://stackoverflow.com/questions/1271320/reseting-generator-object-in-python

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值