Python实现Collatz序列代码是否完善？求优化建议

我是Python编程新手，这是我首次在Stack Overflow发帖。请帮忙判断我实现的Collatz序列代码是否完全可靠，感谢。我已测试负数、0、1及字符串输入，代码如下：

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

while True:
    try:
        number = int(input("Enter a positive non-zero integer: "))
        if number <= 0:
            print("Please enter a positive non-zero integer.")
        else:
            collatz(number)
            break
    except ValueError:
        print("Please enter a valid integer.")

代码可靠性分析 #

• 输入校验部分：做得很到位，能正确捕获非整数输入（比如字符串），也会拦截0和负数并给出明确提示，确保只有正整数能进入Collatz计算流程。
• Collatz逻辑部分：核心逻辑完全符合规则——偶数除以2，奇数执行3*n+1，直到结果为1时终止循环，每一步的计算和打印都没问题。即使输入是1，也能按规则输出4→2→1，流程正确。
• 小细节优化建议：如果想让输入1时的流程更简洁，可以在collatz函数开头加一个判断：if number == 1: print(1); return，不过目前的写法也完全符合Collatz序列的定义，不算问题。

总体来说，你的代码已经是可靠的，能正确处理所有合法输入，也对非法输入做了完善的校验。

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