Как разбить один файл на два файла с помощью Python?

Question

У меня есть файл, содержащий некоторые строки кода, за которым следует шаблон строки. Мне нужно написать все перед строкой, содержащей шаблон строки в файле один и все после шаблона строки в файле два:

например. (Файл-содержание)

• codeline 1
• codeline 2
• Строка шаблона
• codeline 3

Вывод должен быть один файл с codeline 1, codeline 2 и подать два с codeline 3.

Я знаком с написанием файлов, но, к сожалению, я не знаю, как определить контент до и после e string pattern.

Answer 1

Если входной файл помещается в память, самым простым решением является использование str.partition():

with open("inputfile") as f: 
    contents1, sentinel, contents2 = f.read().partition("Sentinel text\n") 
with open("outputfile1", "w") as f: 
    f.write(contents1) 
with open("outputfile2", "w") as f: 
    f.write(contents2) 

Это предполагает, что вы знаете точный текст линии, разделяющей две части.

Answer 2

Наивный пример (который не загружает файл в память, как Свен):

with open('file', 'r') as r: 
    with open('file1', 'w') as f: 
     for line in r: 
      if line == 'string pattern\n': 
       break 
      f.write(line) 
    with open('file2', 'w') as f: 
     for line in r: 
      f.write(line) 

Это предполагает, что 'string pattern' происходит один раз в входном файле.

Если шаблон не является фиксированной строкой, вы можете использовать модуль re.

Answer 3

Более эффективный ответ, который будет работать с большими файлами и потребляют ограниченный объем памяти ..

inp = open('inputfile') 
out = open('outfile1', 'w') 
for line in inp: 
    if line == "Sentinel text\n": 
    out.close() 
    out = open('outfile2', 'w') 
    else: 
    out.write(line) 
out.close() 
inp.close() 

Answer 4

Вам нужно что-то вроде:

def test_pattern(x): 
    if x.startswith('abc'): # replace this with some exact test 
     return True 
    return False 

found = False 
out = open('outfile1', 'w') 
for line in open('inputfile'): 
    if not found and test_pattern(line): 
     found = True 
     out.close() 
     out = open('outfile2', 'w') 
    out.write(line) 
out.close() 

заменить линию StartsWith с тестом, который работает с вашим шаблоном (с использованием сопоставления шаблонов из re, если необходимо, но все, что найдет линия devider).

Answer 5

with open('data.txt') as inf, open('out1.txt','w') as of1, open('out2.txt','w') as of2: 
    outf = of1 
    for line in inf: 
     if 'string pattern' in line: 
      outf = of2 
      continue # prevent output of the line with "string pattern" 
     outf.write(line) 

будет работать с большими файлами, так как он работает по строкам. Предполагается, что string pattern происходит только один раз во входном файле. Мне нравится str.partition() подход лучше если весь файл помещается в память (которая не может быть проблемой)

Использование with гарантирует, что файлы автоматически закрываются, когда вы сделали, или исключение встречается.

Answer 6

Этот подход похож на Lev's, но использует itertools, потому что это весело.

dont_break = lambda l: l.strip() != 'string_pattern' 

with open('input') as source: 
    with open('out_1', 'w') as out1: 
     out1.writelines(itertools.takewhile(dont_break, source)) 
    with open('out_2', 'w') as out2: 
     out2.writelines(source) 

Вы можете заменить функцию dont_break регулярным выражением или чем-либо еще, если это необходимо.

Answer 7

не более трех линий:

with open('infile') as fp, open('of1','w') as of1, open('of2','w') as of2: 
    of1.writelines(iter(fp.readline, sentinel)) 
    of2.writelines(fp)