Python Cookbook 1.4 找到最大或最小的 N 個元素

當我們需要在一個集合,或者列表中找到最大或最小的 N 個元素,這裏我們用到的是 heapq 函數。 1、列表 heapq 模塊中有兩個函數 nlargest 和 nsmallest,可以接受一個數字 n 和列表作爲參數,表示取其中最大或最小的 n 個數字。 返回的是一個已經排好序的列表,包含最大或者最小的幾個元素。 2、字典列表 heapq 還可以通過添加 key 參數作用於字典等更復雜的數據結構
相關文章
相關標籤/搜索