Python中如何在条件满足时终止for循环优化素数求和效率？

Python内置break关键字可以直接终止当前层级的for/while循环，满足迭代到特定条件就退出的需求，针对你的素数求和场景，直接在内层循环判定到非素数时加break即可立刻停止无效遍历。

原有代码的性能问题 #

你写的素数求和逻辑有多处可优化的点：

• 内层循环找到i的因数、将Prime设为False后没有及时退出，仍然会遍历完3到i-1的所有奇数，做了大量无意义计算
• 素数判定的遍历范围冗余：因数是成对出现的，只需要遍历到i的平方根就可以完成判定，不需要遍历到i本身
• 内层判断的i%2==0是无效逻辑：外层循环i从3开始步长为2，所有遍历到的i都是奇数，不可能被2整除
• 素数标记位的重置逻辑可以简化：不需要在循环末尾判断Prime状态再改回True，直接在每次外层循环开始时初始化标记位即可，减少逻辑分支

修改后的代码 #

import math

until = 2000000
total = 2
for i in range(3, until, 2):
    is_prime = True
    # 遍历上限设为根号i，只遍历奇数进一步减少计算量
    for t in range(3, int(math.sqrt(i)) + 1, 2):
        if i % t == 0:
            is_prime = False
            # 找到因数立刻终止内层循环
            break
    if is_prime:
        total += i
print(total)

补充说明 #

• break仅会跳出当前所在的一层循环，不会影响外层循环的正常执行，刚好适配你两层嵌套循环的结构
• 如果需要一次性跳出多层循环，可以通过标记位传递状态，或者把循环逻辑封装成函数，用return直接返回结果
• 如果要计算千万级以上范围的素数和，可以替换为埃氏筛、欧拉筛实现，运行效率会比逐数遍历判定高数十倍

内容的提问来源于stack exchange，提问作者Youp