60D6 – dict 和 set 的 15 个经典使用例子

13. top n 键

找出字典前 n 个最大值,对应的键。

导入 Python 内置模块 heapq 中的 nlargest 函数,获取字典中的前 n 个最大值。

key 函数定义按值比较大小:

In [82]: from heapq import nlargest

In [83]: def topn_dict(d, n):
    ...:     return nlargest(n, d, key=lambda k: d[k])

In [84]: topn_dict({'a': 10, 'b': 8, 'c': 9, 'd': 10}, 3)
Out[84]: ['a', 'd', 'c']
页面: 1 2 3 4 5 6

发表评论

您的电子邮箱地址不会被公开。 必填项已用*标注