Python 数组中最高频率数字的高效查找与处理-Python教程-PHP中文网

Python 数组中最高频率数字的高效查找与处理

心靈之曲

发布： 2025-07-23 15:10:13

原创

673人浏览过

python 数组中最高频率数字的高效查找与处理

本文旨在探讨如何在Python中高效地查找数组中出现频率最高的数字，并处理当多个数字频率相同时，优先返回数值更大的数字这一特殊需求。文章将分析常见实现中的性能瓶颈，并提供基于collections.defaultdict的优化方案，以及不使用defaultdict的替代实现，确保在处理大规模数据时保持卓越性能。

问题定义与挑战

在处理数据时，我们经常需要统计元素出现的频率。一个常见的需求是找出数组中出现次数最多的数字。更进一步，如果存在多个数字拥有相同的最高频率，我们需要一个明确的规则来决定返回哪一个，例如，返回这些数字中数值最大的一个。

一个直观的思路是遍历数组，统计每个数字的出现次数，然后找出频率最高的数字。然而，不恰当的实现方式可能会导致严重的性能问题，尤其是在处理大型数据集时。

低效的实现方式及其分析

考虑以下一种尝试解决此问题的初始实现：

def highest_rank_inefficient(arr):
    count_num = {}
    for i in arr:
        # 这里的逻辑存在问题，并且 arr.count(i) 是性能瓶颈
        if i not in count_num:
            count_num[i] = 0 # 首次遇到时，这里初始化为0是不正确的，应为1或直接赋值
        else:
            count_num[i] = arr.count(i) # 每次循环都重新计算整个列表的count
    # 修正后的max函数，用于处理频率相同取最大值的情况
    return max(count_num, key=lambda x: (count_num.get(x), x))

登录后复制

上述代码中存在两个主要问题：

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

计数逻辑错误： if i not in count_num: count_num[i] = 0 之后没有正确地将当前元素的计数设为1。
性能瓶颈： 最关键的问题在于 count_num[i] = arr.count(i)。arr.count(i) 会遍历整个列表 arr 来计算 i 的出现次数。如果在循环中对每个元素都执行一次 arr.count()，那么对于一个长度为 N 的列表，总时间复杂度将达到 O(N^2)。这意味着当列表长度增加时，执行时间会呈平方级增长，导致在处理中等规模（例如10,000个元素）的数据时变得非常缓慢。

为了演示其低效性，即使修复计数逻辑并正确处理频率相同取最大值的情况，其性能依然堪忧：

def highest_rank_slow(arr):
    count_num = {}
    for i in arr:
        count_num[i] = arr.count(i) # 仍然是 O(N^2) 的瓶颈
    # 使用lambda表达式作为key，优先比较频率，频率相同则比较数字大小
    return max(count_num, key=lambda x: (count_num.get(x), x))

# 示例：
# highest_rank_slow([9, 48, 1, 8, 44, 45, 32]) # 返回 48

登录后复制

对于包含10,000个随机数字的列表，highest_rank_slow 可能需要数秒才能完成。

高效的解决方案：单次遍历与动态更新

为了解决 O(N^2) 的性能问题，我们需要一种只遍历数组一次（O(N) 时间复杂度）就能完成计数和最高频率数字查找的方法。这可以通过维护一个计数字典和一个当前最高频率数字及其频率的变量来实现。

怪兽AI数字人

数字人短视频创作，数字人直播，实时驱动数字人

查看详情

方案一：使用 collections.defaultdict

collections.defaultdict 是一个非常方便的工具，它允许我们为字典中不存在的键提供一个默认值（例如，整数的默认值是0），从而简化计数逻辑。

from collections import defaultdict

def highest_rank(arr):
    """
    查找数组中频率最高的数字，若频率相同则返回数值更大的数字。
    时间复杂度：O(N)，空间复杂度：O(K)，其中 K 是数组中不重复数字的数量。
    """
    count = defaultdict(int)  # 使用 defaultdict(int) 自动将新键的默认值设为 0

    # 初始化最高频率数字和其频率
    # 注意：如果数组为空或只包含0，需要考虑初始值。
    # 这里假设数组非空，且数字可能为0。
    # 为了正确处理所有情况，特别是当数组中所有数字都为负数或0时，
    # highest_rank 应初始化为数组中的第一个元素或一个足够小的值，
    # highest_rank_cnt 初始化为0。

    # 更好的初始化方式，确保第一次迭代能正确设置 highest_rank
    highest_rank_val = None
    highest_rank_cnt = 0

    for num in arr:
        count[num] += 1  # 每次遇到数字，其计数加1
        current_cnt = count[num] # 获取当前数字的最新频率

        # 更新最高频率数字的逻辑：
        # 1. 如果当前数字的频率更高，则更新。
        # 2. 如果当前数字的频率与最高频率相同，但当前数字的数值更大，则更新。
        if highest_rank_val is None or \
           current_cnt > highest_rank_cnt or \
           (current_cnt == highest_rank_cnt and num > highest_rank_val):

            highest_rank_val = num
            highest_rank_cnt = current_cnt

    return highest_rank_val

# 示例测试
print(highest_rank([9, 48, 1, 8, 44, 45, 32]))  # 输出: 48
print(highest_rank([1, 2, 2, 3, 3, 3]))         # 输出: 3
print(highest_rank([1, 2, 2, 3, 3]))           # 输出: 3 (频率相同，3 > 2)
print(highest_rank([5, 5, 5, 1, 1, 1]))         # 输出: 5 (频率相同，5 > 1)
print(highest_rank([]))                         # 输出: None (或根据需求处理空数组)
print(highest_rank([0, 0, 1, 1]))               # 输出: 1

登录后复制

代码解析：

我们使用 defaultdict(int) 创建一个字典 count，其默认值为 0。这意味着当访问一个不存在的键时，它会自动创建该键并赋值为 0，然后我们就可以直接对其进行加法操作。
highest_rank_val 和 highest_rank_cnt 分别存储当前找到的最高频率数字和其对应的频率。
在遍历数组时，我们首先更新当前数字的计数。
然后，我们检查当前数字的频率 (current_cnt) 是否比 highest_rank_cnt 更高。
如果更高，则直接更新 highest_rank_val 和 highest_rank_cnt。
如果频率相同 (current_cnt == highest_rank_cnt)，则进一步检查当前数字 (num) 是否比 highest_rank_val 更大。如果是，也进行更新。
这种方法只需要遍历数组一次，因此时间复杂度为 O(N)。

方案二：不使用 defaultdict 的实现

如果你不想使用 defaultdict，也可以通过标准的字典和 if/else 语句来实现相同的逻辑，但代码会稍微冗长一些：

def highest_rank_no_defaultdict(arr):
    """
    查找数组中频率最高的数字，若频率相同则返回数值更大的数字（不使用 defaultdict）。
    """
    count = {}
    highest_rank_val = None
    highest_rank_cnt = 0

    for num in arr:
        # 手动处理键是否存在
        if num not in count:
            cnt = 1
        else:
            cnt = count[num] + 1
        count[num] = cnt # 更新计数

        # 更新最高频率数字的逻辑与 defaultdict 版本相同
        if highest_rank_val is None or \
           cnt > highest_rank_cnt or \
           (cnt == highest_rank_cnt and num > highest_rank_val):

            highest_rank_val = num
            highest_rank_cnt = cnt

    return highest_rank_val

# 示例测试
print(highest_rank_no_defaultdict([9, 48, 1, 8, 44, 45, 32])) # 输出: 48

登录后复制

虽然这种方法功能上等同于使用 defaultdict 的版本，但在实际性能测试中，defaultdict 通常会略微快一些，因为它在底层进行了优化，避免了显式的 if num not in count 检查。