Sieve of Eratosthenes

erato.py

"""
Sieve of Eratosthenes
"""
import math

def erato_bool(n: int):
    """
    Returned list of primary numbers form 2 to n-1
    """
    if n <= 2:
        return [2, 3]
    seq = [True] * n
    edge = int(math.sqrt(n)) + 1
    for i in range(2, edge):
      for j in range(i * 2, n, i):
        seq[j] = False
    return [i for i in range(n) if seq[i]][2:]

def erato_zero(n: int):
    """
    Returned list of primary numbers form 2 to n-1
    """
    if n <= 2:
        return [2, 3]
    seq = list(range(n))
    edge = int(math.sqrt(n)) + 1
    for i in range(2, edge):
        for j in range(i * 2, n, i):
            seq[j] = 0
    return [i for i in seq if i][1:]