Prime Generator Function

PrimeGenFunc.py

#Returns a list of True and False indicating if a number is prime, for example list[7] = True, list[8] = False
def list_primality(n):
  result = [True] * (n + 1)
  result[0] = result[1] = False
  for i in range(int(math.sqrt(n)) + 1):
    if result[i]:
      for j in range(2 * i, len(result), i):
        result[j] = False
  return result

#Returns all the prime numbers less than or equal to n, in ascending order.

def list_primes(n):
  return [i for (i, isprime) in enumerate(list_primality(n)) if isprime]