cython

1
# distutils: language=c++
2

3

4
from libcpp.vector cimport vector
5

6
def primes(unsigned int nb_primes):
7
    cdef int n, i
8
    cdef vector[int] p
9
    p.reserve(nb_primes)  # allocate memory for 'nb_primes' elements.
10

11
    n = 2
12
    while p.size() < nb_primes:  # size() for vectors is similar to len()
13
        for i in p:
14
            if n % i == 0:
15
                break
16
        else:
17
            p.push_back(n)  # push_back is similar to append()
18
        n += 1
19

20
    # If possible, C values and C++ objects are automatically
21
    # converted to Python objects at need.
22
    return p  # so here, the vector will be copied into a Python list.
23