如何在文件中插入行

Question

我想在文件中插入一行。 如下所示，一旦出现“ AAA”，请插入“ 111”

原始文件

AAA
BBB
CCC
AAA
DDD
AAA

我希望结果是：

AAA
111
BBB
CCC
AAA
111
DDD
AAA
111

这是我的代码

with open(outfile, 'r+') as outfile:
        for line in outfile:    

            if line.startswith('AAA'):
                outfile.write('111\n')
                outfile.flush() 

但是事实证明，每次我运行python时，都只在文件末尾添加“ 111”，而不是仅在“ AAA”开始的后面添加，如下所示。

AAA
BBB
CCC
AAA
DDD
AAA111

关于此有一些问题，但未正确回答。 我真的很想知道那些下注者，您有什么问题吗？ 还是你听不懂？ 还是仅仅因为一些小错误而不会影响问题？

Answer 1

要使用文件fileinput模块就地更新文件，

import fileinput

for line in fileinput.input(outfile, inplace=True):
    if line.strip() == 'AAA':
        print line,
        print 111
    else:
        print line, 

输出： -

AAA
111
BBB
CCC
AAA
111
DDD
AAA
111

Answer 2

r+添加到文件末尾。 这不是您想要的。 只需将代码拆分为读写即可：

filepath = "/file.txt"
with open(filepath, 'r') as infile:
    content = infile.readlines()

out = []
with open(filepath 'w') as outfile:
        for line in content:
            out.append(line)
            if line.startswith('A'):
                out.append("111\n")
        outfile.write("".join(out))

Answer 3

with open("in.txt") as f:
    lines = f.readlines()
    for ind, line in enumerate(lines):
        if line.rstrip() == "AAA":
            lines[ind] = line.rstrip() + "\n111\n"
    with open("in.txt","w") as f1:
        f1.writelines(lines)

输出：

AAA
111
BBB
CCC
AAA
111
DDD
AAA
111

如果要写入新文件并避免将所有文件读入内存，请执行以下操作：

with open("in.txt") as f,open("output.txt","w") as f1:
    for  line in f:
        if line.rstrip() == "AAA":
            f1.write(line.rstrip() + "\n111\n")
        else:
            f1.write(line)