python - 如何在python代码中设置多个输入文件

Question

我正在使用此代码在单个输入文件 (input.txt) 中搜索 target_string，并在输出文件 (output.txt) 中“提取”带有 target_string 的那些行。现在我想执行相同的过程，但有几个输入文件，例如 input1.txt、input2.txt、input3.txt ...

如何修改此代码以执行此操作？

from collections import deque
input_file = 'input.txt' 
output_file = 'output11.txt' 
buscado = 'TCGCCATCCGAATTCCA'

contexto = deque([], 4)  # for keeping the last 4 lines


with open(input_file) as f_in, open(output_file, "w") as f_out:
  # Un bucle for que itere por `f_in` recuperará una línea de cada vez
  for line in f_in:
    contexto.append(line)       
    if  len(contexto) < 4:      
      continue
    if buscado in contexto[1]:  
      f_out.writelines(contexto) 

有人有什么建议吗？我已经挣扎了几个小时 :C

Answer 1

考虑使用 fileinput 模块。

import fileinput
from collections import deque
output_file = 'output11.txt' 
buscado = 'TCGCCATCCGAATTCCA'

contexto = deque([], 4)  # for keeping the last 4 lines


with open(output_file, "w") as f_out:
    for line in fileinput.input(files=["input1.txt", "input2.txt"]):
        contexto.append(line)       
        if len(contexto) < 4:      
            continue
        if buscado in contexto[1]:  
            f_out.writelines(contexto)