Python code for exploring the Collatz conjecture

collatz.py

# Individual Collatz computation
def f(n):
    if n % 2 == 0: 
        return n//2
    else:
        return 3*n+1

# Create sequence of integers from the Collatz function 
# This assumes the Collatz conjecture is true LOL 
def collatz(n): 
    result = [ ] 
    while n != 1: 
        result.append(n)
        n = f(n)
    result.append(1)
    return result