主要介绍了Python找到最大或最小的N个元素实现方法,涉及Python基于heapq模块进行集合运算的相关操作技巧,需要的朋友可以参考下

本文实例讲述了python找到最大或最小的N个元素实现方法。分享给大家供大家参考，具体如下： 问题：想在某个集合中找出最大或最小的N个元素 解决方案：heapq模块中的nlargest()和nsmallest()两个函数正是我们需要的。 >>> import heapq >>> nums=[1,8,2,23,7,-4,18,23,42,37,2] >>> print(heapq.nlargest(3,nums)) [42, 37, 23]