问题: 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