列表的查找模式

Question

我正在编写一个函数来计算一组数字的模式。

如果输入[52、99、37、86、99、99、9999、99，37、37、37]，则输出应[37、99]。正如您所看到的，应该首先使用较小的数字，但我的代码不会这样做。有人能修改我的代码吗？

def mode(L):
    most = max(list(map(L.count, L)))
    return list(set(filter(lambda x: L.count(x) == most, L)))

Answer 1

你在这里使它在计算上相当昂贵。一个<代码>。计数(..)需要线性时间，使得这个算法是二次的。

您可以在此使用计数器对列表执行一次遍历，然后获取最常见的元素，如：

from collections import Counter
from operator import itemgetter
from itertools import groupby

def mode(L):
    _, common = next(groupby(Counter(L).most_common(), itemgetter(1)))
    return sorted(map(itemgetter(0), common))

给定列表中的元素可以有效地散列，这将在线性时间内运行。

Answer 2

< code>sorted()对列表进行排序。

def mode(L):
    most = max(list(map(L.count, L)))
    return sorted(list(set(filter(lambda x: L.count(x) == most, L))))

更新
说明：这是一种非常低效的计算模式的方法。在其他答案中还有性能更高的解决方案。这个答案狭隘地集中在OP提出的问题上。请勿在生产环境中使用此代码。
另请参阅有关此代码的其他改进的评论中的注释。

Answer 3

另一种解决方案是使用< code >集合。计数器

from collections import Counter

nums = [52, 99, 37, 86, 99, 99, 99, 37, 37, 37]

c = Counter(nums)
highest_freq = max(c.values())
mod = [n for n, freq in sorted(c.items()) if freq == highest_freq]

print(mod)

输出：

[37, 99]

如果你只需要一个项目，你也可以只使用:

nums = [52, 99, 37, 86, 99, 99, 99, 37, 37, 37]
c = Counter(nums)
print(max(c))

打印内容：

99