python cook book 读书笔记

最新推荐文章于 2021-09-18 16:51:16 发布

耀东--winter

最新推荐文章于 2021-09-18 16:51:16 发布

阅读量835

点赞数

CC 4.0 BY-SA版权

分类专栏： python

本文链接：https://blog.youkuaiyun.com/haoxuezhe1988/article/details/78979250

python 专栏收录该内容

7 篇文章

订阅专栏

本文介绍如何使用Python的heapq模块快速找到序列中的最大或最小N个元素，并利用collections.Counter类统计序列中各元素出现的频次，进而找出出现次数最多的元素。通过实例演示了这些工具的具体应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1 查找最大或最小的 N 个元素

heapq 模块有两个函数：nlargest() 和 nsmallest() 可以完美解决这个问题。

import heapq
nums = [1, 8, 2, 23, 7, -4, 18, 23, 42, 37, 2]
print(heapq.nlargest(3, nums)) # Prints [42, 37, 23]
print(heapq.nsmallest(3, nums)) # Prints [-4, 1, 2]

heap = list(sums)

heapq.heapify(heap) #进行堆排序

heapy.heappop(heap) #弹出heap里最小值问题
怎样找出一个序列中出现次数最多的元素呢？
解决方案
collections.Counter 类就是专门为这类问题而设计的， 它甚至有一个有用的 most_common() 方法直接给了你答案。
为了演示，先假设你有一个单词列表并且想找出哪个单词出现频率最高。你可以这样做：
words = [
    'look', 'into', 'my', 'eyes', 'look', 'into', 'my', 'eyes',
    'the', 'eyes', 'the', 'eyes', 'the', 'eyes', 'not', 'around', 'the',
    'eyes', "don't", 'look', 'around', 'the', 'eyes', 'look', 'into',
    'my', 'eyes', "you're", 'under'
]
from collections import Counter
word_counts = Counter(words)
# 出现频率最高的3个单词
top_three = word_counts.most_common(3)
print(top_three)
# Outputs [('eyes', 8), ('the', 5), ('look', 4)]