我如何在python中使用返回函数在循环中获得已定义的函数？

Question

我已经编写了python代码来生成所有小于p小于n_max的梅森质数q的列表。使用print时此功能有效，但是我需要使用return，但这会停止循环并仅输出第一个mersenne质数3。

def primes(n):
    i, p, ps, m = 0, 3, [2], n // 2
    sieve = [True] * m
    while p <= n:
        if sieve[i]:
            ps.append(p)
            for j in range((p*p-3)//2, m, p):
                sieve[j] = False
        i, p = i+1, p+2
    return ps

def lucas_lehmer(p):
    if p == 2: return True
    m, i, s = pow(2,p) - 1, 3, 4
    while i <= p:
        i, s = i+1, (pow(s,2) - 2) % m
    return s == 0

def mersenne_prime(n_max):
    L1 = []
    for p in primes(n_max):
        if lucas_lehmer(p):
            q=(2**p)-1
            L1.append(q)
            return L1

任何人都可以给出解决方案，以使代码返回q=2**p-1中的p小于n_max的所有梅森素数吗？

Answer 1

您需要初始化一个列表a以存储所有素数。然后将q附加到它。

def mersenne_prime(n_max):
    a=[]
    for p in primes(n_max):
        print(p)
        if lucas_lehmer2(p):
            q=(2**p)-1
            a.append(a)
    return a