Редактирование конкретной строки в текстовом файле на 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   51  

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))