collatz.py

# Juhani Kaukoranta 16.3.2019
# Collatzin jonon luvut tallennetaan numpy-taulukkoon y
import numpy as np

def Collatz(n):
    x = np.array([n])
    while n > 1:
        if n % 2 == 0:
            n = n//2
        else:
            n = 3*n + 1
 # lisätään saatu luku n numpy-taulukkoon x
        x = np.append(x,n)
#  Collatz-jono on  valmiina palautettavaksi
    return x