Cao Lin's Blog.

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

Word count: 84Reading time: 1 min
2020/09/20 Share

heapq中有两个元素nlargest()nsmallest()可以实现以上功能。

如果 N 等于1时,使用min()max()可以有更快的速度。

如果 N 的大小和集合大小接近的时候,通常先排序这个集合然后再使用切片操作会更快点 (sorted(items)[:N]或者是 sorted(items)[-N:])。

CATALOG