從生成器或清單中取得前N 個元素的優雅方法
在.NET 的LINQ 中,從集合中擷取前N 個項目是使用簡潔的Take() 方法很簡單。想知道如何在 Python 中實現這項壯舉?
輕鬆切片列表
處理列表時,切片語法佔據主導地位:
top5 = array[:5]
這個簡單的表達式有效地獲取數組的前五個元素。如果需要,您可以調整參數以提取特定範圍。
使用 islice 導航產生器
然而,生成器提出了不同的挑戰。直接在 Python 中對它們進行切片是徒勞的。輸入itertools.islice(),您值得信賴的切片生成器盟友:
import itertools top5 = itertools.islice(my_list, 5) # secure the first five elements
生成器的巧妙警告
請記住,在生成器上使用islice() 會部分耗盡它。為了保持整個生成器完好無損,請考慮先將其轉換為元組或清單:
result = tuple(generator)
以上是如何在 Python 中從生成器和清單中檢索前 N 個元素?的詳細內容。更多資訊請關注PHP中文網其他相關文章!