如何用Python检测文本中指定词组并判断其归属列表

需求说明 #

编写函数实现：检测输入文本中是否存在指定词组，并返回该词组所属的列表。

示例数据 #

my_list1 = ["Peter Parker", "Eddie Brock"]
my_list2 = ["Harry Potter", "Severus Snape", "Dumbledore"]

示例演示 #

• 输入文本："Harry Potter was very sad"
• 正确输出：my_list2（注：你给出的示例输出有误，Harry Potter实际属于my_list2）

函数实现代码 #

def check_phrase_list(text, list1, list2):
    # 检查list1中的词组是否存在于文本
    for phrase in list1:
        if phrase in text:
            return "my_list1"
    # 检查list2中的词组是否存在于文本
    for phrase in list2:
        if phrase in text:
            return "my_list2"
    # 无匹配词组时返回提示
    return "No matching phrase found"

# 测试示例
my_list1 = ["Peter Parker", "Eddie Brock"]
my_list2 = ["Harry Potter", "Severus Snape", "Dumbledore"]
test_text = "Harry Potter was very sad"
print(check_phrase_list(test_text, my_list1, my_list2))  # 输出: my_list2

函数说明 #

• 函数接收三个参数：待检测文本text、两个词组列表list1和list2
• 优先匹配list1中的词组，匹配成功直接返回my_list1
• list1无匹配时，再匹配list2，成功则返回my_list2
• 两个列表都无匹配时，返回无匹配提示

内容的提问来源于stack exchange，提问作者Dogukan Yılmaz