Python 如何一次读取N行文件

在本文中，我们将介绍如何使用Python一次读取多行文件的方法。读取多行文件是一种常见的操作，特别适用于处理大型数据文件或者需要按行批量处理的任务。通过一次读取多行文件，我们可以提高程序的效率，并减少磁盘访问的次数。

方法1：使用readlines()函数

readlines()函数是Python内置的文件对象方法，可以一次性读取文件的所有行，并返回一个包含所有行的列表。我们可以使用切片操作符（[:N]）来截取列表的前N行作为我们想要读取的行数。下面是一个示例：

with open('filename.txt', 'r') as f:
    lines = f.readlines()[:N]
    for line in lines:
        print(line)

在上面的示例中，我们使用with语句打开文件，并使用readlines()函数读取文件的所有行。然后，我们使用切片操作符[:N]来截取前N行，并将它们存储在lines列表中。最后，我们使用for循环遍历lines列表，并打印每一行。

方法2：使用迭代器和islice()函数

另一种读取多行文件的方法是使用迭代器和islice()函数。islice()函数是Python内置的itertools模块中的函数，可以通过指定开始和结束的索引来截取迭代器的一部分。我们可以使用islice()函数来截取文件迭代器的前N行。下面是一个示例：

from itertools import islice

with open('filename.txt', 'r') as f:
    lines = islice(f, N)
    for line in lines:
        print(line)

在上面的示例中，我们使用with语句打开文件，并使用islice()函数截取文件迭代器的前N行。然后，我们使用for循环遍历lines迭代器，并打印每一行。

方法3：使用yield和next()函数

最后一种读取多行文件的方法是使用yield关键字和next()函数。yield关键字用于定义生成器函数，可以将一个函数封装成一个迭代器，通过调用next()函数来逐个返回元素。我们可以使用yield关键字和next()函数结合实现一次读取多行文件的功能。下面是一个示例：

def read_lines(filename, N):
    with open(filename, 'r') as f:
        while True:
            lines = [next(f) for _ in range(N)]
            if not lines:
                break
            yield lines

for lines in read_lines('filename.txt', N):
    for line in lines:
        print(line)

在上面的示例中，我们定义了一个生成器函数read_lines()，它接受一个文件名和一个整数N作为参数。在函数内部，我们使用while循环和next()函数来逐个读取文件的每一行，并将它们存储在lines列表中。当lines列表为空时，表示文件已经读取完毕，我们通过break语句跳出循环。然后，我们使用yield关键字返回lines列表。最后，我们使用两层for循环来遍历生成器函数返回的lines列表，并打印每一行。