如何在 Python 中编辑文本文件？

Question

text = open('samiam.txt', 'r+')
keyword = " i "
keyword2 = "-i-"
replacement = " I "
replacement2 = "-I-"

for line in text:    
    if keyword in line:
        text.write(line.replace(keyword, replacement))
        print line
    elif keyword2 in line:
        text.write(line.replace(keyword2, replacement2))
        print line
    else:
        print line
text.close()

我不完全确定为什么没有将文本写入文件。帮忙？

Answer 1

在您的代码中只需替换该行

for line in text:

与

for line in text.readlines():

请注意，这里我假设您尝试在文件末尾添加输出。读取整个文件后，文件指针位于文件末尾（即使您以r+ 模式打开文件）。因此，执行写入实际上会在当前内容之后写入文件末尾。

您可以通过在不同行嵌入text.tell() 来检查文件指针。

这是另一种方法：

with open("file","r") as file: 
    text=file.readlines() 
i=0 
while i < len(text): 
    if keyword in text[i]: 
        text[i]=text[i].replace(keywork,replacement) 
with open("file","w") as file: 
    file.writelines(text)

Answer 2

使用fileinput module。 （另见the MOTW site。）

import fileinput
keyword1 = " i "
keyword2 = "-i-"
replacement1 = " I "
replacement2 = "-I-"
for line in fileinput.input('your.file', inplace=True):
    line = line.rstrip()
    if keyword1 in line:
        line = line.replace(keyword1, replacement1)
    elif keyword2 in line:
        line = line.replace(keyword2, replacement2)
    print line

fileinput 模块，当您使用 inplace 选项时，会重命名输入文件并将标准输出重定向到具有原始文件名的新文件

如果你想保留每一行右边的空白，不要rstrip，使用print line,（注意最后的逗号）输出处理后的行。

Answer 3

这是因为您试图同时读取和写入。尝试类似：

# Read operation
lines = []
for line in text.readlines():
    if keyword in line:
        lines.append(line.replace(keyword, replacement))
        print line
    elif keyword2 in line:
        lines.append(line.replace(keyword2, replacement2))
        print line
    else:
        lines.append(line)
        print line
text.close()

# Write operation
text = open('dummy.py', 'w+')
text.writelines(lines)
text.close()

奖金：

with open('data.txt') as f:
     data = f.read()

更pythonic。

Answer 4

最好使用两个描述符——一个用于读取，另一个用于写入，用于可读性和单次写入操作。

text_read = open('samiam.py', 'r').read()
words_replacer_dict = {" i " : " I ", "-i-" : "-I-"}
replaced_text = ""

for line in text_read.split("\n"):
    for word, new_word in words_replacer_dict.items():
        replaced_text += line.replace(word, new_word)

text_read.close()
text_write = open('samiam.py', 'w')
text_write.write(replaced_text)
text_write.close()

如果您在这种情况下对内存感到困扰，您甚至可以保持计数并根据计数进行写入。只需使用生成器表达式以读取模式打开文件并保持引用计数以满足写入操作。 注意：阅读here（非官方链接），以更好地了解字典方法。

但是，如果您更喜欢使用读写，请始终使用查找操作来获取要替换的行，并在完成文件写入后使用刷新。但是，您不能通过 seek 和 flush 方法替换文件中已经存在的行。您只能在文件中添加一些内容。 （例如）

text = open('samiam.py', 'r+')
count = 1
new_text = ""
for line in text:
    new_text += "%d:%s\n" % (count, line)
    count += 1
text.seek(0)
text.truncate()
text.write(new_text)
text.seek(0)
for line in text:
    print line
text.close()

要更好地了解为什么无法按照您的意愿读写文件，请参阅here