Python实现Collatz序列时异常捕获后如何返回输入循环

核心疑问解答 #

不需要设计复杂的if嵌套逻辑，要实现「输入非法后自动回到输入提示行」的效果，最简洁的方案是用while True无限循环包裹输入校验逻辑：只要用户输入无法通过整数校验，就打印错误提示，自动进入下一轮循环重新展示输入提示；只有输入合法时，才执行break跳出输入循环，进入后续的Collatz计算流程。

现有代码待修正问题 #

• Collatz序列用递归实现存在栈溢出风险：Python默认递归深度上限约为1000，当输入值对应序列长度超过上限时会直接报错，用循环实现更稳定
• 异常处理存在bug：代码中调用了sys.end()但未导入sys模块，且sys模块不存在end()方法，正确的退出方法为sys.exit()
• 输入校验仅执行一次：用户输入非整数时程序打印错误后直接退出，无法回到输入环节
• 逻辑冗余：判断完偶数分支后，剩余情况必然是奇数，不需要额外写number % 2 == 1的判断条件
• Ctrl+C捕获范围不全：原代码仅能捕获输入阶段的中断信号，计算序列过程中按Ctrl+C会直接抛出异常栈，无法干净退出

优化后完整代码 #

import sys

def collatz(number):
    while number != 1:
        if number % 2 == 0:
            number = number // 2
        else:
            number = 3 * number + 1
        print(number)

if __name__ == "__main__":
    try:
        # 无限循环做输入校验，合法才跳出
        while True:
            entry = input('Enter any number: ')
            try:
                number = int(entry)
                break
            except ValueError:
                print('Enter integers only.')
        print(number)
        collatz(number)
    except KeyboardInterrupt:
        # 打印换行避免终端提示符和^C显示在同一行
        print("\n程序已退出")
        sys.exit(0)

优化点说明 #

• 输入逻辑极简：两层try块分工明确，内层try负责校验单次输入格式，错误时仅打印提示不退出程序，自动回到输入行；外层try统一捕获全流程的Ctrl+C中断信号，不管是输入阶段还是计算阶段按Ctrl+C都能干净退出，无异常栈抛出
• Collatz函数改为循环实现，无递归深度风险，代码行数更少，执行效率更高
• 移除冗余判断逻辑，分支更清晰
• 退出时打印换行，终端显示更整洁

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