python实现获取序列中最小的几个元素

本文实例讲述了python实现获取序列中最小的几个元素。分享给大家供大家参考。

具体方法如下：

    import heapq 
    import random 
    def issorted(data): 
     data = list(data) 
     heapq.heapify(data) 
     while data: 
      yield heapq.heappop(data) 

    alist = [x for x in range(10)] 
    random.shuffle(alist) 
    print 'the origin list is',alist 
    print 'the min in the list is' 
    for x in issorted(alist): 
     print x,

程序运行结果如下：

    the origin list is [2, 3, 4, 9, 8, 5, 1, 6, 0, 7]
    the min in the list is
    0 1 2 3 4 5 6 7 8 9

使用了heapq模块和random模块.heapq二叉树，常用来处理优先级序列问题。

此外还有一个更为简单的方法：

print heapq.nsmallest(3,alist)　#打印出alist列表中最小的三个元素最小,如果是字母就是按字母序比较

感兴趣的朋友可以测试运行本文实例，相信本文所述对大家Python程序设计的学习有一定的借鉴价值。