如何使用 re.sub 函数进行以下操作？

Question

我想将以下几行替换为，

ABC+::A1
CDE+:A1A1A1

预期输出：

ABC+::A1
TIM+++2020:11:21:12:51+ZZZ
ABC++ADDEDTIME
CDE+:A1A1A1

我想在上述两行之间添加以下几行。 请注意，从以上几行来看，A1/A1A1A1 是一个会不断变化的变量，但前三个字母字符保持不变。 因此，无论冒号 (+::) 后面提到什么，都必须添加以下两行，

要添加的行

TIM+++2020:11:21:12:51+ZZZ
ABC++ADDEDTIME

Answer 1

我们可以像下面这样很好地利用 re：

import re
regex = r"^(ABC\+::[A-Z0-9]+\n)(?=^CDE\+:[A-Z0-9]+)"
test_str = ("ABC+::A1\n"
    "CDE+:A1A1A1")
subst = "\\1TIM+++2020:11:21:12:51+ZZZ\\nABC++ADDEDTIME\\n"

result = re.sub(regex, subst, test_str, 0, re.MULTILINE)
if result:
    print(result)

正则表达式演示

基本上，正则表达式模式匹配 ABC+::ALPHANUM{1, or more} 后跟 CDE+:ALPHANUM{1, or more} 并替换括号中的第一个匹配项，并在其后添加换行符和新字符串。