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語句對夫婦的選擇，但我不能夠建立邏輯fully.Please幫助

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) 

Answer 2

如何像

>>> with open("pokemonhunt", "r") as f: 
... Lines = [] 
... numlines = 5 
... count = -1 
... for line in f: 
...  if "pokemon" in line:   % gotta catch 'em all!!! 
...  count = 0 
...  if -1 < count < numlines: 
...  Lines.append(line) 
...  count += count 
...  if count == numlines: 
...  with open("pokebox","w") as tmp: tmp.write("".join(Lines)) 

還是我誤解你的問題？