Python输出一个正整数的Collatz序列！

编写函数collatz()，带一个正整数x作为输入参数，输出从x开始的Collatz序列。

res = []
def collatz(x):
    if x == 1:
        return res
    if (x & 1) == 0:
        res.append(x // 2)
        return collatz(x // 2)
    else:
        res.append(3 * x + 1)
        return collatz(3 * x + 1)

x = int(input())
print(collatz(x))