我想迭代两个不同的迭代器,如下所示:
file1 = open('file1', 'r')
file2 = open('file2', 'r')
for item in one_then_another(file1, file2):
print item
我希望先打印file1的所有行,然后再打印file2的所有行。
我想要一些通用的东西,因为迭代器可能不是文件,这只是一个例子。我知道我可以这样做:
for item in [file1]+[file2]:
但这会将两个文件都读到内存中,我希望避免。
from itertools import chain
for line in chain(file1, file2):
pass
fileinput
模块还提供了类似的功能:
import fileinput
for line in fileinput.input(['file1', 'file2']):
pass
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句