无法理解函数行为 - isprime()

Question

我已经构建了两个函数。第一个（不重要但相关，因为它在第二个中调用）告诉一个数字是否是素数：

def is_prime(i):
    if i == 1:
        print("prime")
    if i == 2:
        print("not prime")
    for d in range(2, i):
        if i % d != 0:
            d = d+1
            if d == i:
                print('prime')
                break
        if i % d == 0:
            print('not prime')
            break

我希望能够让这个函数计算从1 到p 的所有素数。 当我要求它将其附加到列表中时-它会单独返回一个空列表和所有值。

def prime_counting(p):
    list_of_primes = []
    for n in range (p+1):
        if is_prime(n) == "prime":
            list_of_primes.append(n)

我该如何解决这个问题？

Answer 1

_{（注意：我没有检查您的is_prime() 函数的逻辑）}
您的代码中明显的错误是函数is_prime() 没有return 任何东西，它只是prints。您希望它为return True 或False，具体取决于数字是否为质数。将其更改为这样做。

然后你可以检查：

if is_prime(n) == True:
    list_of_primes.append(n)

编辑：如 cmets 中所述（正确），编写此类语句的更“pythonic”方式是：

if is_prime(n):
    list_of_primes.append(n)

由于is_prime() 将返回True 或False（因此无需与它们进行比较）。

Answer 2

你的代码有一些缺陷，但你要问的是：

您正在查看is_prime() 的返回值

if is_prime(n) == "prime":

但is_prime() 不包含任何返回语句。

这是一个稍微修改过的版本，它实际上从is_prime()返回一个值：

def is_prime(i):
    if i == 1:
        return "prime"
    if i == 2:
        return "not prime"

    for d in range(2, i):
        if i % d != 0:
            d = d + 1
            if d == i:
                return "prime"
        if i % d == 0:
            return "not prime"


def prime_counting(p):
    list_of_primes = []
    for n in range(p+1):
        if is_prime(n) == "prime":
            print("added %s" % n)
            list_of_primes.append(n)
    return list_of_primes


if __name__ == '__main__':
    print(prime_counting(3))

如果您改为返回布尔值 return True 或 return False，您将可以这样做：

if is_prime(n):
   do_something()

甚至像这样使用“理解”：

prime_list = [n for n in range(1, p+1) if is_prime(n)]