如何使用 re.sub() 在 Python 中添加字符串的开头？

Question

我在 Python 中使用 RegEx 在文本文件中搜索名册中出现的名称，然后附加一个“！”字符到字符串的开头。例如：

名册 = ["name1," "name2," "name3"]

原始字符串 = "name1 回家了。"

输出字符串 - “!name1 回家了。”

我找到了this thread 关于如何附加到字符串末尾的信息，我成功地用于此目的。我已经修改了 RegEx 以在字符串的开头附加，但没有成功。我的尝试如下 - 有什么建议吗？

with open("File.txt", 'r+') as f:
   s = f.read()
   new_s = re.sub(r'^(.*{}.*)^'.format(re.escape("name1")), lambda g: g.group(0) + "!", s, flags=re.MULTILINE)
   f.seek(0)
   f.write(new_s)

Answer 1

去掉.*s - 匹配太多数据会使你的逻辑比它应该的更复杂，没有充分的理由。你的正则表达式不需要被锚定，你也不需要在这里re.MULTILINE（因为没有一个匹配会跨越多行）。

import re

roster = ["name1", "name2", "name3"]
roster_re = re.compile(r'\b(' + '|'.join(re.escape(s) for s in roster) + r')\b')

with open("File.txt", 'r+') as f:
    new_content = roster_re.sub(lambda s: ('!'+s.group(0)), f.read())
    # Note that this is not a safe way to rewrite a file in place; may corrupt data
    f.seek(0)
    f.truncate()
    f.write(new_content)

请参阅How to safely write to a file?，了解您需要进行哪些更改以避免脚本在运行中失败时损坏您的数据文件（系统遇到不合时宜的重新启动、正在写入的文件服务器出现故障等）。