MATH0011: Mathematical Methods II

Code we wrote during lecture 3: listmaker.py

python

def squares(n):
    "Returns a list of square numbers between 1 and n."
    s = []
    i = 1
    while i**2 < n:
        s.append(i**2)
        i += 1
    return s

def primes(n):
    "Returns a list of prime numbers between 1 and n."
    primes = []
    for i in range(2,n):
        a = True
        for p in primes:
            if i%p == 0:
                a = False
                break
            if p**2 > i:
                break
        if a:
            primes.append(i)
    return primes