如何在使用返回函数时在 python 中获取定义的函数以循环？答案

【问题标题】：How do I get a defined function in python to loop whilst using the return function?如何在使用返回函数时在 python 中获取定义的函数以循环？
【发布时间】：2020-03-05 20:05:01
【问题描述】：

我编写了一个 python 代码来生成所有梅森素数 q 的列表，其中 p 小于 n_max。此函数在使用print 时有效，但我需要使用return，但这会停止循环并仅输出第一个梅森素数 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的所有梅森素数？

【问题讨论】：

append 到一个列表，然后返回该列表，就像您在 primes 函数中所做的那样？

标签： python function loops primes

【解决方案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

【讨论】：