這里的版本是針對的一個class的某一個成員變量進行的:
關于如何定義對象的比較方法,請參考往期文章:python定義對象的比較方法
class province_room_quality_data:def __init__(self, room, quality):self.room = roomself.quality = qualitydef __lt__(self, other):return self.quality < other.quality
topk代碼:
from typing import List
import heapq
# 對province_room_quality_data對象的TopK排序算法
def top_k(arr: List[province_room_quality_data], k: int) -> List[province_room_quality_data]:if k == 0:return list()hp = [item for item in arr[:k]]heapq.heapify(hp)for i in range(k, len(arr)):if hp[0].quality < arr[i].quality:heapq.heappop(hp)heapq.heappush(hp, arr[i])ans = [x for x in hp]return ans
最后可以得到基于quality的TopK大的對象數組,注意這里的對象數組并不是基于quality有序的,所以如果最后需要有序的話還得再執行一下sort
newList = top_k(list, 20)
newList.sort(reverse=True)