Python:从一个文件中选择一系列行并将其复制到另一个文件

Question

我有一个文件，我想选择文件中的某些行并将其复制到另一个文件。

在第一个文件中，我要复制的行包含单词“XYZ”，从这里我想选择接下来的 200 行(包括匹配行)并将其复制到另一个文件。

下面是我的代码

match_1 = 'This is match word'
with open('myfile.txt') as f1:
     for num, line in enumerate(f1, 1):
        if log_1 in line:
            print line

上面的代码引导我到行的开头，我需要从匹配的行中选择 200 行，然后复制、移动到另一个文本文件。

我尝试了几个选项，例如 while 语句，但我无法完全构建逻辑。请帮助

Answer 1

一旦找到匹配项，您就可以在文件对象上使用itertools.islice:

from itertools import islice

# some code before matching line is found
chunk = line + ''.join(islice(f1, 200))

这将消耗迭代器 f1 到接下来的 200 行，因此如果将其放入循环中，循环中的 num 计数可能会不一致。

如果在找到匹配项后不需要文件中的其他行，则可以使用:

from itertools import islice

with open('myfile.txt') as f1, open('myotherfile.txt') as f_out:
     for num, line in enumerate(f1, 1):
        if log_1 in line:
            break
     chunk = line + ''.join(islice(f1, 200))
     f_out.write(chunk)