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

Answer 2

您也可以使用简单的 generator expression ：

来完成

for line in (l for f in (file1, file2) for l in f):
    # do something with line

使用此方法，您可以在表达式中指定一些条件：

for line in (l for f in (file1, file2) for l in f if 'text' in l):
    # do something with line which contains 'text'

上面的例子相当于带有循环的生成器：

def genlinewithtext(*files):
    for file in files:
        for line in file:
            if 'text' in line:
                yield line

for line in genlinewithtext(file1, file2):
    # do something with line which contains 'text'

Answer 3

我认为针对此特定文件问题的最Pythonic方法是使用fileinput模块（因为您需要复杂的上下文管理器或使用open进行错误处理），我要去从Ashwini的例子开始，但添加一些东西。首先，最好使用U标志打开Universal Newlines支持（假设你的Python是用它编译的，大多数是），（r是默认模式，但是显式比隐含更好。如果您正在与其他人合作，最好支持他们为您提供任何格式的文件。

import fileinput

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

这也可以在命令行中使用，因为如果你这样做将需要sys.argv [1：]：

import fileinput

for line in fileinput.input(mode='rU'):
   pass

你会像你一样传递shell中的文件：

$ python myscript.py file1 file2

在Python中，如何迭代一个迭代器然后另一个？

3 个答案: