如何根据布尔值重写Python嵌套for循环：禁用外层且保留内层？

这问题我太懂了！复制内层循环的代码不仅违反DRY原则，后期维护起来也容易出问题——改一处得改两处，简直是埋坑。给你几个更优的方案，按可读性和实用性排序：

方案1：把内层逻辑封装成函数（最推荐） #

直接把重复的内层循环抽成独立函数，不管外层是否启用，都调用这个函数。代码清爽，逻辑分离，后期改内层逻辑只需要动一处：

def handle_inner_logic():
    for j in another_generator_function(args...):
        # do more stuff 这里写内层的所有逻辑

# 控制外层循环的布尔值，假设叫enable_outer
if enable_outer:
    for i in my_generator_function(args...):
        # do stuff... 外层需要执行的逻辑
        handle_inner_logic()
else:
    handle_inner_logic()

这个方案的好处是可读性拉满，谁看都能明白你要做什么，而且完全避免了代码重复，是最符合Pythonic风格的解法。

方案2：用条件迭代器统一外层循环 #

如果不想额外加函数，可以给外层循环构造一个“统一入口”：当禁用外层时，让外层迭代器只返回一个占位值（比如None），这样外层循环只会执行一次，然后内层正常跑。如果外层的do stuff需要依赖i，加个判断跳过就行：

# 构造外层迭代器
outer_iterator = my_generator_function(args...) if enable_outer else [None]

for i in outer_iterator:
    # 只有启用外层时才执行外层的逻辑
    if enable_outer:
        # do stuff...
    # 内层循环始终执行
    for j in another_generator_function(args...):
        # do more stuff

如果觉得用列表[None]不够优雅，可以用itertools.repeat生成单次迭代器，更贴合迭代器的场景：

from itertools import repeat

outer_iterator = my_generator_function(args...) if enable_outer else repeat(None, 1)

for i in outer_iterator:
    if enable_outer:
        # do stuff...
    for j in another_generator_function(args...):
        # do more stuff

这个方案的优点是不用拆分函数，保持了循环的结构，但如果外层逻辑比较复杂，判断语句会有点显眼。

方案3：用生成器表达式合并迭代器（进阶玩法） #

如果外层启用时需要让内层循环和外层的每个i绑定，禁用时只跑一次内层，还可以用生成器表达式把两种情况合并：

# 生成一个包含所有需要执行的外层上下文的迭代器
contexts = my_generator_function(args...) if enable_outer else [()]

for ctx in contexts:
    if enable_outer:
        i = ctx
        # do stuff...
    for j in another_generator_function(args...):
        # do more stuff

这个和方案2本质类似，但用空元组()当占位符，更强调“无上下文”的状态，适合外层逻辑依赖i的场景。

总的来说，方案1是首选，因为它最干净、最容易维护。除非你的场景特别简单，否则不建议复制代码——现在省一分钟，后期可能要花十分钟找bug。

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