列表中的第 n 个重复元素

Question

我对以下解决方案的更多 Pythonic 和高性能方法感兴趣。

def nthFrequent(arr,n):

    d = dict((x, arr.count(x)) for x in set(arr))

    value = sorted(d.values(), reverse=True)
    # Pick nth repeated element
    nthrepeat = value[n-1]

    for (key, val) in d.iteritems():
        if val == nthrepeat:
            return key


a=[1,2,3,4,5,6,7,92,3,2,35,9,2,43,4,9,9,9]

print nthFrequent(a,2)

上面的代码将返回 2，因为 2 在 9 之后重复了 3 次，即 4 次。

我正在寻找更优雅的 lambda 使用方式，但我尝试了以下方法，但没有得到所需的结果。

max(((item, a.count(item)) for item in set(a)), key=lambda k: k[1])[0]

上面的将获得最大重复值，即。 9.

如何获得第二个或第 n 个？

Answer 1

对于 collections.Counter，这非常简单。但是请注意，如果 n 值更改为 3，此解决方案将仅返回 3 或 4 之一，因为在这种情况下会出现平局。

import collections

def nthFrequent(arr,n):
    return sorted([(v, k) for k, v in collections.Counter(arr).items()], reverse=True)[n-1][1]

a = [1,2,3,4,5,6,7,92,3,2,35,9,2,43,4,9,9,9]

print nthFrequent(a,2)

另外值得注意的是：元组列表按元组的 0 索引元素排序。因此，您可以使用带有 (count, value) 的元组并返回该值。排序中不需要 lambda。

如果您真的想在没有导入的情况下执行此操作，那么即使这样实现也会更快：

def nthFrequent3(arr, n):
    d = {}
    for v in arr:
        if v not in d:
            d[v] = 0
        d[v] += 1

    return sorted([(v, k) for k, v in d.items()], reverse=True)[n-1][1]

如果您将来决定使用导入，那么也可以看看 itertools。它也有一些方便的工具

def nthFrequent2(arr, n):
    for i, (value, _) in enumerate(itertools.groupby(sorted(arr))):
        if i == n - 1:
            return value

Answer 2

如果您正在寻找单线，以下应该可以工作：

return sorted(((item, a.count(item)) for item in set(a)), key=lambda k: k[1], reverse=True)[n-1][0]

虽然上面使用了更多 Python 语言功能，但我实际上更喜欢原始代码的可读性。

附带说明，在您的原始代码中，您应该return key，因为您当前正在尝试打印一个不返回值的函数。

如果您关心关系，正如@sberry 提到的，您可以这样做：

count 相同时取最小值：

return sorted(((item, a.count(item)) for item in set(a)), 
    key=lambda k: (k[1], k[0]), reverse=True)[n-1][0]

count 相同时取最大值：

return sorted(((item, a.count(item)) for item in set(a)), 
    key=lambda k: (k[1], -k[0]), reverse=True)[n-1][0]

Answer 3

设N为arr的长度，这一行：

d = dict((x, arr.count(x)) for x in set(arr))

按照 N² 的顺序执行多个步骤。首先，遍历arr 以找到它的唯一元素（最坏的情况是每个元素都是唯一的）。其次，对于每个唯一元素，再次遍历整个列表以计算该元素出现的次数。您的单线解决方案也是 N² 的顺序。

这是很多不必要的重复步骤。您只需要查看arr 的每个元素一次。一步，您可以：

• 检查你是否已经看过这个元素

• 增加该元素的计数器

像这样：

counter = {}

for x in arr:
    if x not in counter:
        counter[x] = 0
    counter[x] += 1

pairs = sorted(counter.iteritems(), key=lambda pair: pair[1], reverse=True)

key, count = pairs[n]
return key

在最坏的情况下，每个元素都是唯一的，由于排序，此代码会按照 N*log(N) 的顺序执行多个步骤。 （通过arr 的顺序是N，因为Python 中的dict 查找的顺序是摊销 1。）