Python素数判断：return True语句位置的重要性

本文分析一段Python代码中for循环内return True语句位置差异如何导致素数判断结果错误，并最终影响素数和的计算。

任务：编写两个函数：prime(p)用于判断p是否为素数；primesum(m,n)用于计算区间[m, n]内所有素数的和。

问题：prime(p)函数的两种实现方式：

立即学习“Python免费学习笔记（深入）”；

第一种写法（错误）：

macOS UI 自动化技能包 —— 用 Python 控制你的 Mac

让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac，无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启：辅助功能 + 屏幕录制权限

下载

def prime(p):
    if p == 0 or p == 1:
        return False
    elif p == 2:
        return True
    else:
        for i in range(2, p):
            if p % i == 0:
                return False
            else:
                return True  # 错误：此处return True

第二种写法（正确）：

def prime(p):
    if p == 0 or p == 1:
        return False
    elif p == 2:
        return True
    else:
        for i in range(2, p):
            if p % i == 0:
                return False
        return True  # 正确：此处return True

关键差异：第一种写法将return True放在for循环内部的else块中。这意味着只要循环第一次遇到p % i != 0，函数就立即返回True，而不会检查后续的除数。

例如，当p = 9时，循环首先检查9 % 2，结果不为0；else语句执行，函数立即返回True，错误地将9判断为素数。这导致primesum(1, 10)结果错误，因为9被错误地计入素数和。

第二种写法将return True放在for循环外部。只有当循环完整执行完毕，且没有找到任何能整除p的数时，才会返回True，确保了素数判断的准确性。只有当循环体内的if p % i == 0条件始终为假（p无法被2到p-1之间的任何整数整除）时，函数才会返回True。因此，第二种写法能正确计算区间内素数的和。

总结：return True语句的位置决定了素数判断的完整性。将其置于循环内部会导致提前返回，造成错误判断；将其置于循环外部，则确保了所有可能的除数都被检查，从而得到正确的素数判断结果。