main.py

import time
from math import sqrt, ceil

def checkPrime(num, baseList):    
    for p in baseList:
        if(p > ceil(sqrt(num))): break

        if num % p == 0:
            return False
    return True

def primes(q):
    primelist, number = [2], 3

    while len(primelist) < q:
        if checkPrime(number, primelist):
            primelist.append(number)
        number += 2
    return primelist

def main():
    inicio = time.time()
    r = primes(10)
    print("Saída:", [r[i] for i in [7, 1, 4]])
    fim = time.time()
    print(fim - inicio)

main()