Как удалить несколько строк в текстовом файле с помощью python?

Question

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

Я пытаюсь найти строку, которая соответствует моему поиску и удаляет его, а также следующие 4 строки. У меня есть функция для удаления здесь:

def delete(): 
    del_name = raw_input("What is the first name of the person you would like to delete? ") 
    with open("phonebook.txt", "r+") as f: 
      deletelines = f.readlines() 
      for i, line in enumerate(deletelines): 
       if del_name in line: 
        for l in deletelines[i+1:i+4]: 
         f.write(l) 

Это не работает.

Как удалить несколько записей из текстового файла, подобного этому?

Answer 1

Отвечая на ваш прямой вопрос: вы можете использовать fileinput легко изменить текстовый файл на месте:

import fileinput 
file = fileinput.input('phonebook.txt', inplace=True) 

for line in file: 
    if word_to_find in line: 
     for _ in range(4): # skip this line and next 4 lines 
      next(file, None) 
    else: 
     print line, 

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

Вероятно, лучший ответ: похоже, что вы сделали ручное сериализационное решение. Подумайте о том, чтобы использовать встроенную библиотеку, такую ​​как csv, json, shelve или даже sqlite3, чтобы сохранить ваши данные в удобном для работы формате.