无法修改生成器函数时，如何直接遍历生成器获取指定索引元素？

由于生成器是单向迭代的特性，我们可以通过跟踪当前迭代索引，跳过非目标索引对应的元素的方式，直接操作原生成器获取指定位置的元素，无需借助itertools.compress。这种方式还能避免预先创建掩码数组的内存开销，并且在处理完所有目标元素后提前终止迭代，提升效率。

实现代码 #

def pick_generator_elements(gen, target_indices):
    # 对目标索引排序（生成器只能顺序迭代，必须按索引顺序处理），可选去重避免重复获取
    sorted_targets = sorted(target_indices)
    target_iter = iter(sorted_targets)
    current_target = next(target_iter, None)
    current_idx = 0

    for item in gen:
        if current_idx == current_target:
            yield item
            current_target = next(target_iter, None)
            # 所有目标元素已获取，提前结束迭代
            if current_target is None:
                break
        current_idx += 1

# 测试示例
def mygen(n):
    for k in range(n):
        yield k 

mygenerator = mygen(10)
indices_ls = [0,5,8]

# 直接遍历原生成器获取指定索引元素
for elem in pick_generator_elements(mygenerator, indices_ls):
    print(elem)

关键优势 #

• 无需依赖额外工具库，直接操作原生成器
• 不创建掩码数组，节省内存（尤其当生成器元素数量极大时）
• 处理完所有目标元素后自动终止迭代，避免不必要的生成器遍历
• 支持乱序的目标索引（内部自动排序），也可根据需求保留重复索引（去掉sorted外的set即可）

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