Python中根据布尔输入切换if与while执行逻辑的优雅实现方式

你的需求逻辑清晰，现有实现是正确的，但可以通过合并重复逻辑、简化结构来让代码更优雅，以下是几种可行的优化方案：

方案一：合并循环逻辑（最简洁直观） #

把单次执行和循环执行统一到一个循环里，用标志位控制是否只跑一次：

input_flag: bool

run_once = input_flag
while condition():
    function()
    if run_once:
        break

这个写法的核心是：

• 不管输入是True还是False，先进入循环判断condition()
• 如果是单次模式（run_once=True），执行完function()就直接break退出循环
• 如果是循环模式，就持续执行直到condition()返回False

方案二：用迭代器控制循环次数 #

借助itertools.repeat来生成循环的迭代次数，True时只迭代1次，False时无限迭代：

import itertools

input_flag: bool

for _ in itertools.repeat(None, 1 if input_flag else None):
    if condition():
        function()
    else:
        break  # 单次检查不满足条件就直接退出

这种方式把“执行次数”的逻辑抽离到迭代器里，代码结构更紧凑，适合喜欢函数式风格的场景。

方案三：极简条件控制 #

直接在循环内判断是否需要终止，省去中间变量：

input_flag: bool

while condition():
    function()
    if input_flag:
        break

和方案一逻辑完全一致，只是少了一个中间变量，代码更短，可读性也不受影响。

对比你的原实现，这些方案都避免了重复的if condition(): function()逻辑，让代码结构更统一，可读性更强。

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