如何获得一组具有最大值的键？

Question

我正在处理一个函数

def common_words(dictionary, N):
     if len(dictionary) > N:
         max(dictionary, key=dictionary.get)

函数说明为：

第一个参数是字数字典，第二个参数是 一个正整数 N。这个函数应该更新字典，所以 它包括最常见的（最高频率的词）。最多 N 单词应该包含在字典中。如果包括所有单词 有一些字数会导致字典超过 N 单词，则不应包含具有该字数的单词。 （即，对于第 N+1 个最常用词的平局，省略所有 领带中的单词。）

所以我知道我需要获得具有最高值的 N 个项目，但我不知道该怎么做。我也知道，一旦我得到 N 个项目，如果有任何重复的值，我需要将它们弹出。

---

例如，给定

k = {'a':5, 'b':4, 'c':4, 'd':1} 

然后

common_words(k, 2) 

应修改k，使其变为{'a':5}。

Answer 1

这是我解决这个问题的算法。

1. 将字典中的数据提取到一个列表中，并按字典值的降序对其进行排序。
2. 清除原始字典。
3. 将排序后的数据分组到具有相同值的组中。
4. 使用排序列表中每个组的所有（键、值）对重新填充字典，如果这样可以保持字典总大小 N，则返回。

使用标准的itertools.groupby函数可以轻松完成分组操作。

要执行排序和分组，我们需要一个适当的键函数，如groupby、list 和sorted 文档中所述。由于我们需要每个元组的第二项，我们可以使用

def keyfunc(t):
    return t[1]

或

keyfunc = lambda t: t[1]

但使用operator.itemgetter 效率更高。

---

from operator import itemgetter
from itertools import groupby

def common_words(d, n):
    keyfunc = itemgetter(1)
    lst = sorted(d.items(), key=keyfunc, reverse=True)
    d.clear()
    for _, g in groupby(lst, key=keyfunc):
        g = list(g)
        if len(d) + len(g) <= n:
            d.update(g)
        else:
            break

# test

data = {'a':5, 'b':4, 'c':4, 'd':1} 

common_words(data, 4)
print(data)
common_words(data, 2)
print(data)

输出

{'c': 4, 'd': 1, 'b': 4, 'a': 5}
{'a': 5}

Answer 2

我的算法如下

1. 根据值排序的字典中的第一个构建元组列表 从大到小
2. 检查 item[N-1] 是否匹配 item[N] 的值，如果是，则删除 item[N-1] （索引从 0 开始，所以那里是 -1）
3. 最后，将最多 N 个元素的元组列表切片转换回 dict，如果想保留物品顺序，可以在这里改用OrderedDict

如果字典长度小于 N，它将只返回字典原样

def common_words(dictionary, N):
    if len(dictionary) > N:
        tmp = [(k,dictionary[k]) for k in sorted(dictionary, key=dictionary.get, reverse=True)]
        if tmp[N-1][1] == tmp[N][1]:
            N -= 1
        return dict(tmp[:N])
        # return [i[0] for i in tmp[:N]] # comment line above and uncomment this line to get keys only as your title mention how to get keys
    else:
        return dictionary
        # return dictionary.keys() # comment line above and uncomment this line to get keys only as your title mention how to get keys

>>> common_words({'a':5, 'b':4, 'c':4, 'd':1}, 2)
{'a': 5}

OP想在函数内修改输入字典并返回None，可以修改如下

def common_words(dictionary, N):
    if len(dictionary) > N:
        tmp = [(k,dictionary[k]) for k in sorted(dictionary, key=dictionary.get, reverse=True)]
        if tmp[N-1][1] == tmp[N][1]:
            N -= 1
        # return dict(tmp[:N])
        for i in tmp[N:]:
            dictionary.pop(i[0])

>>> k = {'a':5, 'b':4, 'c':4, 'd':1}
>>> common_words(k, 2)
>>> k
{'a': 5}