Python车型搜索函数扩展：多词车型匹配实现求助

作业要求编写一个返回车型列表的函数，需支持不区分大小写的指定车型搜索匹配。当前实现的search_by_model函数仅能匹配单个车型词（例如调用search_by_model("Audi A4,Audi a4 2021,Audi A40", "A4")会返回['Audi A4', 'Audi a4 2021']），但无法处理多词车型的匹配需求（比如搜索"a4 2021"时无法正确匹配对应的车型）。需要修改函数实现多词车型的精准匹配。

原代码如下：

def search_by_model(all_cars: str, model: str) -> list:
    if not all_cars or not model:
        return []
    all_makes = []
    cars = all_cars.split(",")
    for car in cars:
        unique_car = car.split(" ")
        if model.lower() == unique_car[1].lower():
            all_makes.append(" ".join(unique_car[0:]))
    return all_makes

原代码问题分析 #

• 仅提取车型的第二个单词做单词匹配，完全不支持多词搜索场景
• 匹配逻辑是严格的单词相等对比，无法处理多词连续组合的车型匹配

修改后的实现代码 #

def search_by_model(all_cars: str, model: str) -> list:
    if not all_cars or not model:
        return []
    
    target_model = model.lower().strip()
    cars = [car.strip() for car in all_cars.split(",")]
    matched_cars = []
    
    for car in cars:
        car_lower = car.lower()
        target_words = target_model.split()
        car_words = car_lower.split()
        
        # 检查目标词序列是否是车型词序列的连续子部分
        match_found = False
        for i in range(len(car_words) - len(target_words) + 1):
            if car_words[i:i+len(target_words)] == target_words:
                match_found = True
                break
        if match_found:
            matched_cars.append(car)
    
    return matched_cars

关键修改说明 #

1. 统一大小写处理：把搜索词和车型都转为小写，彻底解决大小写不敏感问题
2. 多词连续匹配：将搜索词和车型拆分为单词列表，验证搜索词的单词序列是否是车型单词序列的连续子段，既支持多词匹配，也避免了部分匹配的误判（比如不会把"Audi A40"误判为匹配"A4"）
3. 字符串清洗：对分割后的车型做strip()处理，避免前后空格干扰匹配结果

测试示例 #

• 调用search_by_model("Audi A4,Audi a4 2021,Audi A40", "A4") → 返回['Audi A4', 'Audi a4 2021']（保留原单词匹配功能）
• 调用search_by_model("Audi A4,Audi a4 2021,Audi A40", "a4 2021") → 返回['Audi a4 2021']（新增多词精准匹配）
• 调用search_by_model("BMW 320i M,BMW 320,BMW M3", "320i") → 返回['BMW 320i M']

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