[Python]Collatz 序列

编写一个名为 collatz()的函数:
它有一个名为 number 的参数。如果参数是偶数，那么 collatz()就打印出 number // 2，并返回该值。
如果 number 是奇数，collatz()就打印并返回 3 * number + 1。
然后编写一个程序，让用户输入一个整数，并不断对这个数调用 collatz()，直到函数返回值１
（令人惊奇的是，这个序列对于任何整数都有效，利用这个序列，你迟早会得到 1！ 既使数学家也不能确定为什么。你的程序在研究所谓的“Collatz序列”，它有时候被称为“最简单的、不可能的数学问题”）。
记得将 input()的返回值用 int()函数转成一个整数，否则它会是一个字符串。

代码：

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


print('Enter number：')
try:
    num = int(input())
except ValueError:
    print('请输入一个整数！')
    
number = collatz(num)
while True:
    if number != 1:
        number = collatz(number)
    else:
        break

结果截图：