檢索文件的前N 行

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

使用列表理解讀取前N 行

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

with open(path_to_file) as input_file:
    head = [next(input_file) for _ in range(lines_number)]
print(head)

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

使用 islice() 函數

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

from itertools import islice

with open(path_to_file) as input_file:
    head = list(islice(input_file, lines_number))
print(head)

from itertools import islice&]

使用 open(path_to_file) 作為輸入檔： 頭 = 列表（islice（輸入文件，行數）） print(head)這裡，islice()函數用來迭代輸入檔的前lines_number行，建立提取行的列表。