计算耗时结果少的列表推导式，如何实时输出元素？

当然有办法解决这个问题！你遇到的核心痛点是列表推导式是立即求值的——哪怕你只需要最后10个结果，它也会先把range(100000000)里的所有元素都遍历一遍，全部计算完才返回列表，白白浪费大量时间。

换成生成器表达式（把方括号改成圆括号）就能实现惰性求值，再配合一个装饰器，就能做到计算出一个元素就立即打印一个，不用等全部完成。下面是完整的实现方案：

装饰器代码实现 #

import sys

def print_as_computed(gen):
    try:
        while True:
            # 每次只获取生成器的下一个元素（惰性计算）
            v = next(gen)
            # 打印当前元素
            print(v)
            # 强制刷新输出缓冲区，避免内容攒着不显示
            sys.stdout.flush()
            # 把元素yield出去，方便后续收集完整结果
            yield v
    except StopIteration:
        # 生成器迭代结束时的提示（可选）
        print("\n✅ 所有元素计算完成")
        return

用法示例 #

把原来的列表推导式改成生成器表达式，再用装饰器包装后迭代：

# 生成器表达式：惰性计算，每次只生成一个符合条件的元素
my_generator = (x for x in range(100000000) if x < 10)

# 用装饰器包装生成器，迭代过程中会实时打印元素
# 同时还能把结果收集到列表里
final_result = list(print_as_computed(my_generator))

# 最后可以查看完整结果
print("最终结果列表:", final_result)

为什么这么做有效？ #

1. 生成器的惰性求值：生成器表达式不会一次性计算所有元素，只有当调用next()时才会计算下一个符合条件的元素，避免了遍历1亿次的无用功。
2. 装饰器的实时处理：装饰器每次拿到生成器产出的元素，就立即打印并刷新输出（sys.stdout.flush()是为了防止系统把输出攒在缓冲区里不显示），同时把元素传递出去，不影响后续收集完整结果。

如果你用的是Python 2，只需要把print(v)改成print v即可，其他逻辑完全一致。

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