如何在Python中提取檔案的前N行？

檢索檔案的前N 行

通常，在處理大型原始資料檔案時，有必要提取特定的數字用於進一步處理或分析的線路。在 Python 中，有多種方法可以完成此任務。

使用列表理解讀取前N 行

一個簡單有效的方法涉及利用列表理解：

<code class="python">with open(path_to_file) as input_file:
    head = [next(input_file) for _ in range(lines_number)]
print(head)</code>

此方法使用next( ) 函數迭代輸入文件，並將前lines_number 行儲存在頭列表中。

使用islice() 函數

另一種方法是利用Python 的itertools 模組：

<code class="python">from itertools import islice

with open(path_to_file) as input_file:
    head = list(islice(input_file, lines_number))
print(head)</code>

這裡， islice() 函數用於迭代輸入檔的前lines_number 行，建立提取行的列表。

作業系統的影響

無論使用何種作業系統，上述實作都應該有效。不過，值得注意的是，在 Python 2 中，next() 函數被稱為 xrange()，這可能需要在較舊的程式碼庫中進行相應的調整。

以上是如何在Python中提取檔案的前N行？的詳細內容。更多資訊請關注PHP中文網其他相關文章！