1def primes(int kmax):
2    """
3    >>> primes(20)
4    [2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67, 71]
5    """
6    cdef int n, k, i
7    cdef int[1000] p
8    result = []
9    if kmax > 1000:
10        kmax = 1000
11    k = 0
12    n = 2
13    while k < kmax:
14        i = 0
15        while i < k and n % p[i] <> 0:
16            i = i + 1
17        if i == k:
18            p[k] = n
19            k = k + 1
20            result.append(n)
21        n = n + 1
22    return result
23