Python高效实现多模型到单制造商的映射-Python教程-PHP中文网

Python高效实现多模型到单制造商的映射

碧海醫心

发布： 2025-10-29 11:48:06

原创

939人浏览过

Python高效实现多模型到单制造商的映射

本文探讨了在python中将多个值（如产品型号）映射到单个值（如制造商）的有效策略。通过将原始的制造商-型号列表映射关系转换为型号-制造商的倒排字典，可以显著提高查找效率。文章详细介绍了使用双层循环和字典推导式构建倒排字典的方法，并演示了如何利用此优化结构进行快速数据查询，从而避免了低效的嵌套循环。

在数据处理和分析中，我们经常会遇到需要将一组相关联的多个值映射到它们的唯一分类或属性值的情况。例如，给定一系列产品型号，需要快速识别出每个型号对应的制造商。直接通过遍历查找的方式虽然可行，但当数据量增大时，其性能瓶颈会变得尤为突出。

问题场景描述

假设我们有一个产品型号列表 model_name，以及一个 manufacturer_dict，其中键是制造商名称，值是该制造商生产的型号列表。我们的目标是为 model_name 中的每个型号找到对应的制造商，并生成一个 manufacturer_list。

示例数据：

model_name = ["Legion", "ROG", "Nitro", "TUF", "Predator"]

manufacturer_dict = {
    "ASUS": ["ROG", "TUF"],
    "ACER": ["Predator", "Nitro"],
    "Lenovo": ["Legion"]
}

登录后复制

传统低效的查找方法

一种直观但效率不高的做法是使用嵌套循环。对于 model_name 中的每个型号，我们遍历 manufacturer_dict，检查该型号是否存在于某个制造商的型号列表中。

立即学习“Python免费学习笔记（深入）”；

manufacturer_list = []
for each_model in model_name:
    manufacturer = "" # 初始化制造商为空字符串
    for key, value in manufacturer_dict.items():
        if each_model in value:
            manufacturer = key
            break  # 找到后立即跳出内部循环
    manufacturer_list.append(manufacturer)

print(manufacturer_list)
# 输出: ['Lenovo', 'ASUS', 'ACER', 'ASUS', 'ACER']

登录后复制

这种方法的缺点在于，对于 model_name 中的每一个元素，都需要潜在地遍历 manufacturer_dict 的所有条目及其对应的型号列表。当 model_name 列表和 manufacturer_dict 的规模都很大时，这种重复的查找操作会导致显著的性能开销，时间复杂度较高。

优化策略：构建倒排查找字典

为了提高查找效率，我们可以采用预处理的方式，将原始的 manufacturer_dict 转换为一个更适合快速查找的结构。具体来说，我们可以创建一个新的字典，其中键是每个型号，值是其对应的制造商。我们将这种结构称为“倒排查找字典”或“模型-制造商字典”。

通过这种方式，后续的查找操作将从遍历转变为字典的哈希查找，其平均时间复杂度接近 O(1)，大大提升了效率。

方法一：使用双层循环构建倒排字典

我们可以通过简单的双层循环来遍历 manufacturer_dict，并构建 models_to_manufacturer_dict。

manufacturer_dict = {
    "ASUS": ["ROG", "TUF"],
    "ACER": ["Predator", "Nitro"],
    "Lenovo": ["Legion"]
}

models_to_manufacturer_dict = {}

# 遍历制造商及其型号列表
for manufacturer, models in manufacturer_dict.items():
    # 遍历每个制造商下的所有型号
    for model in models:
        models_to_manufacturer_dict[model] = manufacturer

print(models_to_manufacturer_dict)
# 输出: {'ROG': 'ASUS', 'TUF': 'ASUS', 'Predator': 'ACER', 'Nitro': 'ACER', 'Legion': 'Lenovo'}

登录后复制

方法二：使用字典推导式（更简洁的Pythonic方法）

Python的字典推导式提供了一种更简洁、更具表达力的方式来完成上述任务。

可图大模型

可图大模型（Kolors）是快手大模型团队自研打造的文生图AI大模型

查看详情

manufacturer_dict = {
    "ASUS": ["ROG", "TUF"],
    "ACER": ["Predator", "Nitro"],
    "Lenovo": ["Legion"]
}

models_to_manufacturer_dict = {
    model: manufacturer
    for manufacturer, models in manufacturer_dict.items()
    for model in models
}

print(models_to_manufacturer_dict)
# 输出: {'ROG': 'ASUS', 'TUF': 'ASUS', 'Predator': 'ACER', 'Nitro': 'ACER', 'Legion': 'Lenovo'}

登录后复制

这两种方法都将原始数据转换为一个高效的查找结构。字典推导式通常更受推荐，因为它代码量少，可读性高，且执行效率通常与显式循环相当。

使用优化后的字典进行高效查找

一旦 models_to_manufacturer_dict 构建完成，我们就可以利用它来高效地为 model_name 列表中的每个型号查找制造商。

model_name = ["Legion", "ROG", "Nitro", "TUF", "Predator"]

# 假设 models_to_manufacturer_dict 已经通过上述方法之一构建完成
models_to_manufacturer_dict = {
    'ROG': 'ASUS',
    'TUF': 'ASUS',
    'Predator': 'ACER',
    'Nitro': 'ACER',
    'Legion': 'Lenovo'
}

# 使用列表推导式进行高效查找
manufacturer_list = [models_to_manufacturer_dict.get(model, "未知制造商") for model in model_name]

print(manufacturer_list)
# 输出: ['Lenovo', 'ASUS', 'ACER', 'ASUS', 'ACER']

登录后复制

这里我们使用了字典的 get() 方法，它允许我们为未找到的键提供一个默认值（例如 "未知制造商"），这比直接使用 [] 访问键更健壮，可以避免 KeyError。