Python高效统计字典嵌套列表值在目标列表中的出现次数

本文将介绍如何在python中高效地统计一个字典中嵌套列表的值在另一个目标列表中的总出现次数。针对常见但效率低下的o(n³)嵌套循环方法，文章提出了一种通过预处理目标列表来优化性能的o(n)解决方案，并详细分析其实现原理、代码示例及性能优势。

引言

在Python编程中，我们有时会遇到这样的需求：给定一个字典 my_dict，其键对应的值是一个列表；同时给定一个独立的目标列表 my_list。我们的目标是创建一个新的字典，其中包含 my_dict 的所有键，但对应的值是该键所关联的列表中所有元素在 my_list 中出现的总次数。

例如，考虑以下数据：

my_dict = {'A': ['A', 'B'], 'B': ['C', 'D'], 'C': ['E', 'F']}
my_list = ['A', 'D', 'A', 'C', 'F', 'F']

我们期望的输出是 {'A': 2, 'B': 2, 'C': 2}。这是因为：

• 对于键 'A'，其关联列表为 ['A', 'B']。在 my_list 中，'A' 出现了2次，'B' 出现了0次，总计2次。
• 对于键 'B'，其关联列表为 ['C', 'D']。在 my_list 中，'C' 出现了1次，'D' 出现了1次，总计2次。
• 对于键 'C'，其关联列表为 ['E', 'F']。在 my_list 中，'E' 出现了0次，'F' 出现了2次，总计2次。

原始思路及性能考量

初学者可能会倾向于使用多层嵌套循环来解决这个问题。一种直观但效率不高的思路是：对于 my_dict 中的每个键及其关联列表，遍历该列表中的每个元素，然后针对每个元素再次遍历 my_list 来统计其出现次数。

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

这种方法通常会导致极差的性能。具体来说，其时间复杂度会达到 O(N³)，其中：

1. 最外层循环遍历 my_dict 的键（假设有 K 个键）。
2. 中间层循环遍历每个键对应的嵌套列表中的元素（假设最长嵌套列表有 M 个元素）。
3. 最内层操作是在 my_list 中查找元素是否存在或统计其出现次数。在Python中，对列表使用 in 操作符或遍历查找的平均时间复杂度是 O(N)，其中 N 是 my_list 的长度。

因此，总的迭代次数大致为 K * M * N。在上述示例中，这相当于 3 * 2 * 6 = 36 次基本操作，对于小规模数据尚可接受。但当 K、M、N 变得很大时，O(N³) 的性能会迅速导致程序运行缓慢甚至崩溃。例如，如果 my_list 很大，每次在 my_list 中查找一个元素，都需要遍历 my_list 的大部分内容，这会重复进行大量不必要的计算。

优化方案：预处理与O(N)算法

为了显著提升性能，我们可以采用一种预处理策略，将时间复杂度从 O(N³) 降低到 O(N)。核心思想是：首先对 my_list 进行一次遍历，统计其中每个元素的出现次数，并将其存储在一个字典中。由于字典的查找和插入操作平均时间复杂度为 O(1)，这使得后续的计数汇总变得极其高效。

Quinvio AI

AI辅助下快速创建视频，虚拟代言人

59

查看详情

下面是使用纯Python实现此优化方案的函数：

代码示例

def count_nested_values(my_dict: dict, my_list: list) -> dict:
    """
    高效统计字典嵌套列表值在目标列表中出现的总次数。

    参数:
        my_dict (dict): 键为字符串，值为字符串列表的字典。
        my_list (list): 包含字符串元素的目标列表。

    返回:
        dict: 一个新字典，键为my_dict的键，值为对应元素在my_list中出现的总次数。
    """
    # 步骤1: 预处理my_list，统计每个元素的出现次数
    # 使用字典存储计数，实现O(1)查找
    counts = {}
    for list_val in my_list:
        counts[list_val] = counts.get(list_val, 0) + 1

    # 步骤2: 遍历my_dict，根据预处理的counts字典汇总计数
    new_dict = {}
    for key, associated_list in my_dict.items():
        new_dict[key] = 0  # 初始化当前键的总计数
        # 遍历与当前键关联的列表中的每个元素
        for item_in_associated_list in associated_list:
            # 从预处理的counts字典中获取该元素的出现次数
            # 使用.get()方法，如果元素不存在于my_list中，则默认为0，避免KeyError
            new_dict[key] += counts.get(item_in_associated_list, 0)

    return new_dict

# 示例用法
my_dict = {'A': ['A', 'B'], 'B': ['C', 'D'], 'C': ['E', 'F']}
my_list = ['A', 'D', 'A', 'C', 'F', 'F']

result = count_nested_values(my_dict, my_list)
print(result)

输出:

{'A': 2, 'B': 2, 'C': 2}

性能分析

让我们详细分析上述优化方案的性能：

1. 步骤1: 预处理 my_list

   • 此步骤通过一次循环遍历 my_list 中的所有元素。
   • 对于每个元素，将其作为键存入 counts 字典，并更新其计数。字典的插入和查找操作平均时间复杂度为 O(1)。
   • 因此，此步骤的总时间复杂度为 O(N_list)，其中 N_list 是 my_list 的长度。

2. 步骤2: 遍历 my_dict 并汇总计数

   • 此步骤首先遍历 my_dict 的所有键值对。假设 my_dict 有 K 个键。
   • 对于每个键，它会遍历其关联的嵌套列表中的所有元素。假设所有嵌套列表中的元素总数为 N_nested_values。
   • 在内部循环中，我们使用 counts.get(item, 0) 来获取元素的出现次数。由于 counts 是一个字典，此查找操作的平均时间复杂度为 O(1)。
   • 因此，此步骤的总时间复杂度为 O(K + N_nested_values)。

综合以上两步，整个算法的整体时间复杂度为 O(N_list + K + N_nested_values)。在最坏情况下，这可以简化为 O(N)，其中 N 是输入数据（my_list 长度和 my_dict 中所有元素的总数）的最大规模。

与 O(N³) 的原始方法相比，O(N) 的算法在处理大规模数据时具有压倒性的性能优势。例如，如果 N 为 1000，O(N³) 意味着 10亿次操作，而 O(N) 仅意味着 1000次操作。

注意事项与总结

• 内存与时间权衡: 这种优化方案通过引入一个中间字典 counts 来存储 my_list 的预处理结果。这意味着会消耗额外的内存空间，用于存储 counts 字典。然而，这种内存开销通常是可接受的，因为它带来了显著的时间性能提升。在大多数实际应用中，时间性能往往比少量内存消耗更受关注。
• 适用场景: 对于输入数据规模较小（例如，my_list 和 my_dict 中的元素数量都在几十或几百以内）的场景，O(N³) 的简单方法可能也能快速完成任务，此时过度优化可能没有必要。但一旦数据规模增长，性能瓶颈就会显现，此时采用 O(N) 算法是至关重要的。
• 可读性: 优化后的代码虽然比最简单的嵌套循环稍长，但其逻辑结构清晰，分为“预处理”和“汇总”两个明确的阶段，使得代码更易于理解和维护。
• Pythonic替代方案: 在Python的标准库 collections 中，Counter 类提供了更简洁的方式

以上就是Python高效统计字典嵌套列表值在目标列表中的出现次数的详细内容，更多请关注php中文网其它相关文章！