是否可以更快地计算列表中数字出现的数量？

Question

我正在尝试编写一个函数来计算列表中数字的出现，并且该顺序是根据数字上升（从列表中的0到最大值）而不是出现。这是我写的功能：

def sort_counts(sample):
    result = []
    for i in range(max(sample)+1):
        result.append(sample.count(i))
    return result

例如：

>>> sort_counts([1,2,2,3,3,4,1,1,1,1,2,5])
>>> [0, 5, 3, 2, 1, 1]

我了解到sample.count如果列表中有更多数字，则可以缓慢工作。我可以编写此功能更快/更简单吗？

Answer 1

收集模块的计数器是计算列表中项目出现数量的好方法

from collections import Counter
lst = [1,2,2,3,3,4,1,1,1,1,2,5]
# create a counter object
c = Counter(lst)
# get the counts
[c[i] for i in range(max(c)+1)]
# [0, 5, 3, 2, 1, 1]

Answer 2

如果您不想使用计数器，则可以在创建适当大小的结果数组之后简单地迭代列表中的值：

sample = [1,2,2,3,3,4,1,1,1,1,2,5]
result = [0] * (max(sample)+1)
for v in sample:
    result[v] += 1

输出：

[0, 5, 3, 2, 1, 1]

示例数据的速度明智，如果counter 解决方案是1x时间，这是关于4X和您现有的解决方案，涉及8x。随着列表的越来越长，Counter的速度优势降低，相对性能更像1X，2X和4X代码>。

Answer 3

如果您不想通过样本的元素使用计数并更新结果。它可以减少找到元素计数所需的多次遍历所需的时间。

def sort_counts(sample):
    result = [0]*(max(sample)+1)
    for s in samples:
        result[s]+=1
    return result