为何try-except无法捕获拆分生成器时抛出的StopIteration异常？

错误原因 #

1. 惰性求值导致异常无法捕获：你定义的part是生成器表达式，它不会在test函数的try块内执行next(vid)，而是等到内层遍历（for j in i）时才会触发。此时抛出的StopIteration会被Python自动转换为RuntimeError（PEP 479规定），外层的try-except根本无法捕获这个异常。
2. 剩余元素未被丢弃：当原生成器剩余元素不足size个时，子生成器仍会取出所有剩余元素，而不是直接丢弃。

正确实现 #

要实现“拆分生成器为固定大小块，不足size的部分丢弃”的需求，可以在每次迭代中主动收集满size个元素，中途若遇到StopIteration则终止，且只返回收集满的块：

def chunk_generator(original_gen, chunk_size):
    while True:
        chunk = []
        try:
            for _ in range(chunk_size):
                chunk.append(next(original_gen))
            yield (x for x in chunk)  # 若不需要子生成器，直接yield chunk更简单
        except StopIteration:
            # 收集的元素不足chunk_size，直接丢弃
            break

# 测试代码
res = chunk_generator((i for i in range(100)), 30)
for chunk in res:
    for num in chunk:
        print(num, end=" ")
    print()

说明 #

• 主动循环chunk_size次收集元素，确保只有当收集满chunk_size个元素时才返回对应的子生成器（或列表）。
• 当原生成器耗尽时，except StopIteration触发，此时若chunk长度不足chunk_size，直接终止不返回，实现丢弃剩余元素的效果。

更简洁的写法（利用itertools） #

如果允许使用itertools模块，可以用itertools.islice简化实现：

import itertools

def chunk_generator(original_gen, chunk_size):
    while True:
        chunk = list(itertools.islice(original_gen, chunk_size))
        if len(chunk) < chunk_size:
            break
        yield (x for x in chunk)  # 或直接yield chunk

# 测试
res = chunk_generator((i for i in range(100)), 30)
for chunk in res:
    print(' '.join(map(str, chunk)))

说明 #

• itertools.islice可以从生成器中取出指定数量的元素，若不足则返回剩余的。
• 通过判断取出的chunk长度是否等于chunk_size，不足则终止，自动丢弃剩余元素。

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