python - 在 Python 中，我如何迭代一个迭代器，然后再迭代另一个？

Question

我想迭代两个不同的迭代器，像这样:

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]:

但这会将两个文件都读入内存，我希望避免这样做。

Answer 1

使用 itertools.chain :

from itertools import chain
for line in chain(file1, file2):
   pass

fileinput模块也提供了类似的功能:

import fileinput
for line in fileinput.input(['file1', 'file2']):
   pass