4-array/heapq/queue模塊

forrest23 發(fā)布于2019-07-30 18:45 / 1409人閱讀

摘要：定義了一個非常類似的模塊，其函數(shù)接受兩個參數(shù)，第一個參數(shù)是預(yù)先定義好的類型，第二個參數(shù)，一般為一個序列。很少見到代碼輸出是中實現(xiàn)堆排序的模塊。這里主要看一下優(yōu)先級隊列定義優(yōu)先級比較輸出

array

array 定義了一個非常類似list的模塊，其array 函數(shù)接受兩個參數(shù)，第一個參數(shù)是預(yù)先定義好的類型，第二個參數(shù)，一般為一個序列。很少見到
代碼：

import array

a = array.array("b",b"abcd")
print(a)
print(a[0])

輸出：

array("b", [97, 98, 99, 100])
97

heapq

heapq 是python中實現(xiàn)堆排序的模塊。

from heapq import *
import random
# 創(chuàng)建一個堆排序
data = []
for i in range(10):
    heappush(data,random.randint(1,20))

print(data)

# 使用 heappop 移除最小的元素
small_num = heappop(data)
print(small_num)
print("pop移除最小元素后: ",data)
# 使用heapreplace替換最小元素，會先執(zhí)行pop，然后replace
n = heapreplace(data,100)
print(n)
print("執(zhí)行replace后:",data)

# n個最大 / n個最小
lagest = nlargest(3,data)
small = nsmallest(3,data)

print("3個最大的值：",lagest)
print("3個最小的值：",small)

輸出

[2, 3, 8, 7, 4, 18, 12, 17, 16, 13]
2
pop移除最小元素后:  [3, 4, 8, 7, 13, 18, 12, 17, 16]
3
執(zhí)行replace后: [4, 7, 8, 16, 13, 18, 12, 17, 100]
3個最大的值： [100, 18, 17]
3個最小的值： [4, 7, 8]

queue

隊列，在線程一節(jié)有總結(jié)過，有先進先出隊列，也有優(yōu)先級隊列，隊列結(jié)合線程，可以保證線程之間通信的安全。這里主要看一下優(yōu)先級隊列

from queue import PriorityQueue
import threading
import functools
Q = PriorityQueue()

@functools.total_ordering
class Job:
    def __init__(self,priority,desc):
        self.priority = priority
        self.desc = desc
        return 
    # 定義優(yōu)先級比較
    def __eq__(self,other):
        try:
            self.priority == other.priority
        except AttributeError as e:
            return NotImplemented
    
    def __lt__(self,other):
        try:
            self.priority < other.priority
        except AssertionError:
            return NotImplemented

def worker():
    while not Q.empty():
        item = Q.get()
        import time 
        time.sleep(1)
        print(item.desc)
    
    Q.all_tasks_done


if __name__ == "__main__":
    Q.put(Job(3,"mid job"))
    Q.put(Job(10,"important job"))
    Q.put(Job(1,"low job"))
    ts = [threading.Thread(target=worker),threading.Thread(target=worker)]
    for t in ts:
        t.start()
    
    t.join()

輸出