Python优雅迭代长度非步长整数倍的缓冲区的方法

Python标准库中并没有专门针对这种“带余数场景”的range变体，但可以通过更简洁的方式实现需求，避免冗余的min写法重复出现。

方案1：通用生成器函数封装 #

如果需要频繁处理这类场景，封装一个生成器函数是最优雅的方式，比每次手动计算更简洁：

def chunk_indices(total_length, step):
    start = 0
    while start < total_length:
        current_size = min(step, total_length - start)
        yield start, current_size
        start += step

使用时直接迭代生成器即可，无需重复计算：

buf = b"..." * 953  # 示例缓冲区
for start, size in chunk_indices(len(buf), 33):
    # 将索引和长度传入目标函数
    process_buffer(buf, start, size)

方案2：简化单次迭代写法 #

如果只是单次使用，不需要封装函数，可以用更紧凑的表达式替代冗余的min写法（本质逻辑一致，但更简洁）：

buf_length = len(buf)
step = 33
for start in range(0, buf_length, step):
    size = step if start + step <= buf_length else buf_length - start
    # 调用函数处理
    process_buffer(buf, start, size)

为什么没有标准库实现？ #

Python的range设计目标是生成均匀步长的整数序列，不关联具体的长度限制（比如缓冲区长度），因此不会内置处理余数的逻辑。但通过上述封装或简化写法，完全可以满足通用场景的需求，且代码可读性和复用性都能得到保证。

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