Редактирование конкретной строки в текстовом файле на python
Допустим у меня есть текстовый файл, содержащий:
Dan
Warrior
500
1
0
есть ли способ я могу изменить определенную строку в текстовом файле? Сейчас у меня есть это:
#!/usr/bin/env python
import io
myfile = open('stats.txt', 'r')
dan = myfile.readline()
print dan
print "Your name: " + dan.split('n')[0]
try:
myfile = open('stats.txt', 'a')
myfile.writelines('Mage')[1]
except IOError:
myfile.close()
finally:
myfile.close()
Да, я знаю, что myfile.writelines('Mage')[1] некорректно. Но ты ведь понимаешь, что я имею в виду? Я пытаюсь отредактировать строку 2, заменив воина на мага. Но могу ли я это сделать?
6 ответов:
вы хотите сделать что-то вроде этого:
# with is like your try .. finally block in this case with open('stats.txt', 'r') as file: # read a list of lines into data data = file.readlines() print data print "Your name: " + data[0] # now change the 2nd line, note that you have to add a newline data[1] = 'Mage\n' # and write everything back with open('stats.txt', 'w') as file: file.writelines( data )причина этого в том, что вы не можете сделать что-то вроде "изменить строку 2" непосредственно в файле. Вы можете только перезаписать (не удалять) части файла - это означает, что новое содержимое просто покрывает старое содержимое. Итак, если вы написали " маг "над строкой 2, результирующая строка будет "Mageior".
вы можете использовать fileinput для редактирования на месте
import fileinput for line in fileinput.FileInput("myfile", inplace=1): if line .....: print line
def replace_line(file_name, line_num, text): lines = open(file_name, 'r').readlines() lines[line_num] = text out = open(file_name, 'w') out.writelines(lines) out.close()и затем:
replace_line('stats.txt', 0, 'Mage')
Если ваш текст содержит только один человек:
import re # creation with open('pers.txt','wb') as g: g.write('Dan \n Warrior \n 500 \r\n 1 \r 0 ') with open('pers.txt','rb') as h: print 'exact content of pers.txt before treatment:\n',repr(h.read()) with open('pers.txt','rU') as h: print '\nrU-display of pers.txt before treatment:\n',h.read() # treatment def roplo(file_name,what): patR = re.compile('^([^\r\n]+[\r\n]+)[^\r\n]+') with open(file_name,'rb+') as f: ch = f.read() f.seek(0) f.write(patR.sub('\1'+what,ch)) roplo('pers.txt','Mage') # after treatment with open('pers.txt','rb') as h: print '\nexact content of pers.txt after treatment:\n',repr(h.read()) with open('pers.txt','rU') as h: print '\nrU-display of pers.txt after treatment:\n',h.read()Если ваш текст содержит несколько человек:
импорт re
# creation with open('pers.txt','wb') as g: g.write('Dan \n Warrior \n 500 \r\n 1 \r 0 \n Jim \n dragonfly\r300\r2\n10\r\nSomo\ncosmonaut\n490\r\n3\r65') with open('pers.txt','rb') as h: print 'exact content of pers.txt before treatment:\n',repr(h.read()) with open('pers.txt','rU') as h: print '\nrU-display of pers.txt before treatment:\n',h.read() # treatment def ripli(file_name,who,what): with open(file_name,'rb+') as f: ch = f.read() x,y = re.search('^\s*'+who+'\s*[\r\n]+([^\r\n]+)',ch,re.MULTILINE).span(1) f.seek(x) f.write(what+ch[y:]) ripli('pers.txt','Jim','Wizard') # after treatment with open('pers.txt','rb') as h: print 'exact content of pers.txt after treatment:\n',repr(h.read()) with open('pers.txt','rU') as h: print '\nrU-display of pers.txt after treatment:\n',h.read()Если "работа" человека имеет постоянную длину в тексте, вы можете изменить только часть текста, соответствующую" работе " желаемого человека: это та же идея, что и у сендерле.
но по-моему, лучше было бы поставить характеристики лиц в диктофон записана в файл с cPickle:
from cPickle import dump, load with open('cards','wb') as f: dump({'Dan':['Warrior',500,1,0],'Jim':['dragonfly',300,2,10],'Somo':['cosmonaut',490,3,65]},f) with open('cards','rb') as g: id_cards = load(g) print 'id_cards before change==',id_cards id_cards['Jim'][0] = 'Wizard' with open('cards','w') as h: dump(id_cards,h) with open('cards') as e: id_cards = load(e) print '\nid_cards after change==',id_cards
вы можете сделать это двумя способами, выбрать то, что подходит вашему требованию:
Метод И.) замена с помощью номера строки. Вы можете использовать встроенную функцию
enumerate()в этом случае:сначала в режим чтения получить все данные в переменной
with open("your_file.txt",'r') as f: get_all=f.readlines()во-вторых, запись в файл (где перечислить приходит в действие)
with open("your_file.txt",'w') as f: for i,line in enumerate(get_all,1): ## STARTS THE NUMBERING FROM 1 (by default it begins with 0) if i == 2: ## OVERWRITES line:2 f.writelines("Mage\n") else: f.writelines(line)II способ.) используя ключевое слово, которое вы хотите заменить:
открыть файл в режим чтения и скопируйте содержимое в список
with open("some_file.txt","r") as f: newline=[] for word in f.readlines(): newline.append(word.replace("Warrior","Mage")) ## Replace the keyword while you copy."воин" был заменен на "Маг", поэтому запишите обновленные данные в файл:
with open("some_file.txt","w") as f: for line in newline: f.writelines(line)это выход будет в обоих случаях:
Dan Dan Warrior ------> Mage 500 500 1 1 0 0
сегодня вечером я практиковался в работе с файлами и понял, что могу построить ответ Йохена, чтобы обеспечить большую функциональность для повторного/многократного использования. К сожалению, мой ответ не решает проблему работы с большими файлами, но облегчает жизнь в небольших файлах.
with open('filetochange.txt', 'r+') as foo: data = foo.readlines() #reads file as list pos = int(input("Which position in list to edit? "))-1 #list position to edit data.insert(pos, "more foo"+"\n") #inserts before item to edit x = data[pos+1] data.remove(x) #removes item to edit foo.seek(0) #seeks beginning of file for i in data: i.strip() #strips "\n" from list items foo.write(str(i))