Поиск и замена строки под определенной строкой

Question

Как я могу выполнить поиск и ЗАМЕНИТЬ строку. Чтобы быть более конкретным.

У меня есть текстовый файл с

SAMPLE 
AB 
CD 
.. 
TYPES 
AB 
QP 
PO 
.. 
RUNS 
AB 
DE 
ZY 

Я хочу заменить AB с XX, только под линиями SAMPLE и RUNS. Я уже пробовал несколько способов использования replace(). Я попробовал что-то вроде

if 'SAMPLE' in line: 
    f1.write(line.replace('testsample', 'XX')) 
if 'RUNS' in line: 
    f1.write(line.replace('testsample', 'XX')) 

и что не работал.

Answer 1

Файл итератор над линиями в Python:

for line in file: 
    output.write(line) # save as is 
    if 'SAMPLE' in line or 'RUNS' in line: 
     line = next(file, "") # move to the next line 
     output.write(line.replace('AB', 'XX')) # save replacing AB with XX 

Для поддержки SAMPLE/БЕГА линии, которые следуют друг за другом SAMPLE/БЕГА линии, например:

SAMPLE 
SAMPLE 
AB 

вы можете:

for line in file: 
    output.write(line) # save as is 
    while 'SAMPLE' in line or 'RUNS' in line: 
     line = next(file, "") # move to the next line 
     output.write(line.replace('AB', 'XX')) # save replacing AB with XX 

Answer 2

Самый простой способ - перебрать файл по строкам, и каждый раз, когда вы видите строку SAMPLE или RUNS, чтобы сохранить флаг, означающий «предыдущая строка была той, которую я искал». Любая другая строка сбросит этот флаг. Теперь на каждой итерации вы проверяете, был ли флаг установлен во время предыдущей итерации, и если это вы делаете свой .replace.

Answer 3

Предполагая, что вы не хотите, чтобы линии ниже SAMPLES заменены, и вы не хотите заменять строка, которая содержит ABC, это будет немного более надежным, если ваш файл содержит больше текста на строку, чем строки, которые вы ищете.

valid_sections = set(('RUNS', 'SAMPLE')) 
for raw_line in fobj_in: 
    clean_line = raw_line.strip() 
    if clean_line in valid_sections: 
     fobj_out.write(raw_line) 
     raw_line = next(fobj_in) 
     clean_line = raw_line.strip() 
     if clean_line == 'AB': 
      fobj_out.write(raw_line.replace('AB', 'XX')) 
     else: 
      fobj_out.write(raw_line) 
    else: 
     fobj_out.write(raw_line)