Редагування конкретного рядка в текстовому файлі на Python


86

Скажімо, у мене є текстовий файл, що містить:

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, замінивши Warrior на Mage. Але чи можу я це навіть зробити?


1
Я думаю, що ця публікація охоплює те, що ви шукаєте: stackoverflow.com/questions/1998233/…
Кайл Уайлд

1
Якщо вам доводиться робити такі речі багато, можливо, ви захочете перетворити цей файл з тексту на щось на зразок bdb або іншого bdb-подібного.
Нік Бастін,

Відповіді:


123

Ви хочете зробити щось подібне:

# 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" безпосередньо у файлі. Ви можете перезаписувати (а не видаляти) частини файлу - це означає, що новий вміст просто охоплює старий вміст. Отже, якщо ви написали "Mage" над рядком 2, отриманий рядок буде "Mageior".


2
Привіт Йохен, вираз "з відкритим (ім'я файлу, режим)" також неявно закриває ім'я файлу після виходу програми, правда?
Раду

@Gabriel Thx, це важливо зазначити, хоча я все ще не використовую with ... як файл. Пітонічний чи ні, мені це не подобається :)
Раду,

@Radu це питання звикання. Раніше я також закривав відкриті файли вручну через, close.але тепер я бачу, що використання withблоку набагато чистіше.
Габріель,

5
Чи правильно вважати, що це рішення найкраще для невеликих файлів? В іншому випадку нам може знадобитися багато пам'яті лише для зберігання даних. Більше того, навіть для 1 редагування нам потрібно написати все це знову.
Arindam Roychowdhury

11
Погано .. а якщо у вас файл розміром 20 Гб?
Brans Ds

21

Ви можете використовувати введення файлів для редагування на місці

import fileinput
for  line in fileinput.FileInput("myfile", inplace=1):
    if line .....:
         print line

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

9
це завантажило б весь вміст файлу в пам’ять, що може бути не дуже добре, якщо файл величезний.
Steve Ng

@SteveNg у вас є рішення проблеми, яку ви помітили? і ця відповідь, і прийнята покладаються на завантаження всього файлу в пам’ять
Blupon

13

Ви можете зробити це двома способами, вибрати те, що відповідає вашим вимогам:

Спосіб I.) Заміна за допомогою номера рядка. У 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.  

"Warrior" замінено на "Mage", тому запишіть оновлені дані у файл:

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           

3

Якщо ваш текст містить лише одну особу:

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

Якщо ваш текст містить декілька осіб:

імпортувати повторно

# 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

2

Сьогодні ввечері я практикував роботу над файлами і зрозумів, що можу спиратися на відповідь Йохена, щоб забезпечити більшу функціональність для багаторазового / багаторазового використання. На жаль, моя відповідь не стосується питання роботи з великими файлами, але полегшує життя в менших файлах.

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

0

Це найпростіший спосіб зробити це.

fin = open("a.txt")
f = open("file.txt", "wt")
for line in fin:
    f.write( line.replace('foo', 'bar') )
fin.close()
f.close()

Сподіваюся, це спрацює у вас.


-1

# читати рядки файлів та редагувати певний елемент

файл = відкритий ("pythonmydemo.txt", 'r')

a = file.readlines ()

друк (a [0] [6:11])

a [0] = a [0] [0: 5] + 'Ericsson \ n'

друк (a [0])

файл = відкритий ("pythonmydemo.txt", 'w')

file.writelines (a)

file.close ()

друк (а)


1
Ласкаво просимо до Stack Overflow! Зверніть увагу, що ви відповідаєте на дуже давнє, на яке вже відповідали. Ось посібник із відповіді .
help-info.de

-2

Припустимо, у мене є файл із таким іменем file_name:

this is python
it is file handling
this is editing of line

Ми повинні замінити рядок 2 на "модифікація виконана":

f=open("file_name","r+")
a=f.readlines()
for line in f:
   if line.startswith("rai"):
      p=a.index(line)
#so now we have the position of the line which to be modified
a[p]="modification is done"
f.seek(0)
f.truncate() #ersing all data from the file
f.close()
#so now we have an empty file and we will write the modified content now in the file
o=open("file_name","w")
for i in a:
   o.write(i)
o.close()
#now the modification is done in the file
Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.