如何基于预定义句子列表实现文本的自定义拆分？

需求：实现一个函数，按照预定义的自定义句子列表拆分文本，这些句子可包含任意特殊字符。

示例输入 #

text = "My name. is A. His name is B. Her name is C. That's why..."
delims = ["My name. is", "His name is", "Her name is"]

期望函数效果 #

def custom_sentence_split(text, delims):
     # 实现逻辑
     return result

custom_sentence_split(text, delims)
# 返回结果：["My name. is", "  A. ", "His name is", "  B. ", "Her name is", " C. That's why..."]

初始非简洁实现 #

之前有一个可行但不够简洁的实现：

def collect_output(text, finds):
    text_copy = text[:]
    retn = []
    for found in finds:
        part1, part2 = text_copy.split(found, 1)
        retn += [part1, found]
        text_copy = part2
    return retn


def custom_sentence_split(text, splitters):
    pattern = "("+"|".join(splitters)+"|)"
    finds = list(filter(bool, re.findall(pattern, text)))
    output = collect_output(text, finds)
    return output

简洁最优实现 #

最终找到更简洁的可行方案：利用正则的re.split方法，通过re.escape处理分隔符中的特殊字符（避免正则语法冲突），同时用分组保留分隔符本身：

import re

def custom_sentence_split(text, delims):
    pattern = "(" + "|".join(map(re.escape, delims)) + ")"
    # 过滤掉空字符串，匹配结果可能开头出现空值
    return list(filter(bool, re.split(pattern, text)))

# 测试
text = "My name. is A. His name is B. Her name is C. That's why..."
delims = ["My name. is", "His name is", "Her name is"]
print(custom_sentence_split(text, delims))
# 输出：["My name. is", "  A. ", "His name is", "  B. ", "Her name is", " C. That's why..."]

内容的提问来源于stack exchange，提问作者Vova