在Python中讀取檔案的前N行

在資料處理中，通常只需要操作大型資料集的一部分。這就是從文字檔案中讀取指定行數的能力發揮作用的地方。

Python 的內建方法

Python 提供了一種直接的讀取方法檔案的前N 行：

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

此程式碼片段在指定路徑開啟文件，然後迭代lines_number 次行，將檢索到的行儲存在頭列表中。

作業系統注意事項

作業系統通常不會影響此任務的實現。無論底層系統如何，Python 都會管理文件訪問，從而使程式碼可以跨不同平台移植。

使用islice 的替代方法

讀取前N 行的另一個選項是使用itertools 模組中的islice 函數：

from itertools import islice

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

此方法傳回一個產生器迭代器，該迭代器產生檔案的前lines_number行。 list() 函數用於將生成器轉換為列表，以便於處理。

結論

這些程式碼片段提供了讀取指定行數的可靠方法來自 Python 中的文字檔案。無論您是修剪大型資料集還是在前幾行執行特定操作，這些方法都提供高效且通用的解決方案。