Question

我正在尝试向文件（input.txt）添加多行，并替换同一文件（input.txt）中的多行，但是我的代码仅在文件末尾插入了行。您知道我如何修复代码以获得所需的预期输出文件吗？

我的代码：

import re

searchtext1 = """
AB     3.483e-01   2.52e-02 ; 3.46 0.0123
"""
# add these lines after searchtext1
addtext1 = """
CD     2.123e-01   1.31e-02 ; 7.25 0.0145
DE     4.896e-01   7.25e-02 ; 8.25 0.0185
"""
searchtext2 = """
; atom
#atomnumber

#molecule
[weight]
"""
# replace these lines to searchtext2
changetext2 = """
; iron
#48kcal
35 mol
#12 g
"""

with open('input.txt', 'ab+') as infile:
    matches1 = re.findall(r'^(\d+)\.(.*)$', searchtext1, re.MULTILINE)
    infile.write(addtext1)

    matches2 = re.findall(r'^(\d+)\.(.*)$', searchtext2, re.MULTILINE)
    infile.write(changetext2)

input.txt：

[atom]
123
[bonds]
XY     4.212e-01   4.18e-02 ; 8.01 0.0487
AB     3.483e-01   2.52e-02 ; 3.46 0.0123

[molecule]
1 2
3 4
TY     0.412e-01   1.72e-02 ; 0.32 0.0455

; atom
#atomnumber

#molecule
[weight]
calculated value is 5 kcal/mol
end file

预期的输出文件：

[atom]
123
[bonds]
XY     4.212e-01   4.18e-02 ; 8.01 0.0487
AB     3.483e-01   2.52e-02 ; 3.46 0.0123
CD     2.123e-01   1.31e-02 ; 7.25 0.0145
DE     4.896e-01   7.25e-02 ; 8.25 0.0185

[molecule]
1 2
3 4
TY     0.412e-01   1.72e-02 ; 0.32 0.0455

; iron
#48kcal
35 mol
#12 g
calculated value is 5 kcal/mol
end file

Answer 1

您已以附加模式“ ab +”打开文件，该模式指定所有写入都将到达文件末尾。要以所需方式更改文件的某些部分，必须重写整个文件。

一个常见的模式是创建一个新的临时文件，写入新数据，然后将新文件移到旧文件上。这样，覆盖是原子性的，并且如果程序崩溃，丢失数据的机会就更少。

Answer 2

就像克里斯在评论中提到的那样，我建议尝试使用已经可以与格式交互的库-configparser对我来说似乎是显而易见的选择，但是如果格式不正确，可能会有原因可能与您的问题有所不同。

此外，如果您想使用re，我已经对您的代码进行了一些更新。这是我所做的调整：

从三引号字符串的开头和结尾删除了换行符，因为它们将成为匹配的一部分，而这可能不是您想要的。我个人的喜好是要明确有关实际替换中的换行符。
在搜索文本中转义的[和]，因为它们是re中的特殊字符，会使其混淆（它们指定字符类）。
使用with打开单独的输入和输出文件。当您退出with块时，这些将被清除。如果您真的要替换input.txt，我想您可以在之后将output.txt移到顶部。
您可以使用re.sub直接替换出现的内容，而不是搜索它们，而是在文本中找到索引，然后替换/附加。在添加文本的情况下，我已经将searchtext1和addtext1都替换掉了。（我使用了f弦，但如果愿意，也可以"{search}\n{add}".format(search=searchtext, add=addtext1)来做。
最后，我们将更新后的文本写回到光盘上。

希望有帮助！

import re

searchtext1 = """AB     3.483e-01   2.52e-02 ; 3.46 0.0123"""
# add these lines after searchtext1
addtext1 = """CD     2.123e-01   1.31e-02 ; 7.25 0.0145
DE     4.896e-01   7.25e-02 ; 8.25 0.0185"""

searchtext2 = """; atom
#atomnumber

#molecule
\[weight\]"""
# replace these lines to searchtext2
changetext2 = """; iron
#48kcal
35 mol
#12 g"""

with open('input.txt', 'r') as infile, open("output.txt", "w") as outfile:
    intext = infile.read()

    intext = re.sub(searchtext1, f"{searchtext1}\n{addtext1}", intext)
    intext = re.sub(searchtext2, changetext2, intext)

    outfile.write(intext)

使用findall替换并添加多行

2 个答案: