检索文件的前 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)

这里，islice()函数用于迭代输入文件的前lines_number行，创建提取行的列表。

的效果操作系统

无论使用什么操作系统，上述实现都应该有效。不过，值得注意的是，在 Python 2 中，next() 函数被称为 xrange()，这可能需要在较旧的代码库中进行相应的调整。