如何修改Python正则表达式代码以实现指定字符串格式转换？

算力阿峰

2026-7-10

问题：修改Python正则表达式实现指定格式输出

现有如下Python正则表达式代码，输出结果为['A_B1_C', 'A_B2_C', 'A_D2_C']，但期望得到的结果是['(A)_(B)(1)_(C)', '(A)_(B)(2)_(C)', '(A)_(D)(2)_(C)']，请问应如何修改该代码？

原代码：

import re
texts = ['A_B1_C', 'A_B2_C', 'A_D2_C']
keyword = '(A)_(B|D)(.*)_(C)'
[(re.search(keyword, text).group()) for text in texts]

解决方案

原代码中re.search(...).group()默认返回整个匹配的原始字符串，不会自动将捕获组按你需要的格式重组。你需要手动提取各捕获组内容，再按目标格式拼接：

修改后的代码（循环版）

import re
texts = ['A_B1_C', 'A_B2_C', 'A_D2_C']
keyword = '(A)_(B|D)(.*)_(C)'
result = []
for text in texts:
    match_obj = re.search(keyword, text)
    if match_obj:
        # 按目标格式拼接各捕获组
        formatted_str = f'({match_obj.group(1)})_({match_obj.group(2)})({match_obj.group(3)})_({match_obj.group(4)})'
        result.append(formatted_str)
print(result)

简化版（列表推导式）

import re
texts = ['A_B1_C', 'A_B2_C', 'A_D2_C']
keyword = '(A)_(B|D)(.*)_(C)'
result = [
    f'({m.group(1)})_({m.group(2)})({m.group(3)})_({m.group(4)})'
    for m in (re.search(keyword, t) for t in texts)
    if m
]
print(result)