Python 3中如何按4行一组读取文件直至EOF？

嘿，我懂你这种想要优雅解决问题的心情——之前用列表推导式只能拿前4行，用计数器又觉得不够Pythonic，对吧？刚好我有几个不错的方案，都是利用Python的迭代器特性来实现按4行一组读取文件直到EOF，而且代码简洁还高效。

方案一：用itertools.islice（最简洁的Pythonic方式） #

itertools.islice可以直接对文件对象（本身就是迭代器）进行切片操作，每次取指定数量的行，直到迭代器耗尽。这种方法不需要把整个文件加载到内存，非常适合处理大文件。

import itertools
import os

myfile = "你的文件路径.txt"
if os.path.isfile(myfile):
    with open(myfile, 'r') as f:
        while True:
            # 每次从文件迭代器中取4行，返回列表；空列表表示已到EOF
            chunk = list(itertools.islice(f, 4))
            if not chunk:
                break
            # 在这里处理你的4行数据，比如打印或解析
            print("正在处理一组行：")
            for line in chunk:
                print(line.strip())

为什么这个方法优雅？ #

• 不需要维护额外的计数器变量，逻辑清晰
• 惰性读取，只在需要时读取4行，内存占用低
• 利用Python内置的迭代器工具，符合Python的设计哲学

方案二：自定义生成器函数（复用性更强） #

如果需要在多个地方复用“按n行一组读取文件”的逻辑，写一个生成器函数是更好的选择。它把读取逻辑封装起来，主代码会更干净。

版本1（Python 3.8+，用海象运算符简化） #

import os

def read_file_in_chunks(file_path, chunk_size=4):
    with open(file_path, 'r') as f:
        # 每次尝试读取chunk_size行，直到返回空列表
        while chunk := list(next(f) for _ in range(chunk_size)):
            yield chunk

myfile = "你的文件路径.txt"
if os.path.isfile(myfile):
    for chunk in read_file_in_chunks(myfile):
        # 处理每组数据
        print("当前组的行：")
        print([line.strip() for line in chunk])

版本2（兼容Python 3.8以下） #

如果你的Python版本还没支持海象运算符，可以用try-except捕获StopIteration来处理：

import os

def read_file_in_chunks(file_path, chunk_size=4):
    with open(file_path, 'r') as f:
        while True:
            chunk = []
            try:
                # 尝试读取chunk_size行
                for _ in range(chunk_size):
                    chunk.append(next(f))
            except StopIteration:
                # 读取到文件末尾时，如果还有剩余的行（不足4行），也返回
                if chunk:
                    yield chunk
                break
            yield chunk

myfile = "你的文件路径.txt"
if os.path.isfile(myfile):
    for chunk in read_file_in_chunks(myfile):
        # 处理每组数据
        print("当前组的行：")
        print([line.strip() for line in chunk])

生成器的优势 #

• 逻辑封装，主代码只需要关注如何处理每组数据，不用关心读取细节
• 可以灵活调整chunk_size参数，比如改成按5行一组读取只需要传个参数
• 同样是惰性读取，内存友好

这两种方案都比用计数器+break的方式更优雅，而且能完美处理文件末尾不足4行的情况，直到读取到EOF为止。

内容的提问来源于stack exchange，提问作者DoubleVoid