Python中如何将质数校验函数作为参数实现通用质数计数器

完全可以实现你需要的无重复逻辑的通用计数器，Python 支持将函数作为一等对象传入其他函数，根本不需要为每一种素性校验方法单独编写重复的循环统计代码。

实现思路 #

只需要把重复的计数循环逻辑抽成独立的通用函数，将「质数校验逻辑」「统计数值范围」作为入参传入即可：

• 第一个入参位置接收自定义的质数校验函数，函数内部不需要关心传入的校验逻辑是基于平方根遍历还是费马小定理，只需要统一调用传入的函数对每个数做判断
• 第二、第三个入参位置接收统计范围的起始、结束值，避免把范围硬编码在函数内部，提升后续灵活性

顺便修正你原伪代码里的逻辑小问题：原primsqr函数中if x%i ==0分支内，break和return False的顺序写反了，break会直接跳出循环，后面的return False永远不会被执行到，会导致非质数被误判为质数；同时两个校验函数对x<=1的场景没有显式返回布尔值，默认返回None，长期来看会造成返回值类型不统一的问题。

可直接运行的示例代码 #

# 修正逻辑后的平方根遍历质数校验
def primsqr(x):
    if x > 1:
        for i in range(2, int(x ** 0.5) + 1):
            if x % i == 0:
                return False
        return True
    return False

# 修正逻辑后的费马小定理质数校验
def primeferm(x):
    if x > 1:
        return pow(2, x-1, mod=x) == 1
    return False

# 通用质数计数函数
def count_primes(check_func, range_start, range_end):
    prime_amount = 0
    for num in range(range_start, range_end + 1):
        if check_func(num):
            prime_amount += 1
    return prime_amount

调用方式 #

只需要按照需求传入对应的校验函数和范围值即可，不需要重复写循环逻辑：

# 统计0~3000000范围内两种方法识别出的质数数量
sqrt_check_result = count_primes(primsqr, 0, 3000000)
fermat_check_result = count_primes(primeferm, 0, 3000000)

# 统一计算两种方法的结果差异百分比
max_count = max(sqrt_check_result, fermat_check_result)
diff_percent = abs(sqrt_check_result - fermat_check_result) / max_count * 100

方案优势 #

• 完全消除重复的计数逻辑，后续如果新增其他质数校验方法（如米勒-拉宾素性测试），只需要编写新的校验函数，直接传入通用计数函数即可，不需要改动计数模块的代码
• 支持灵活调整统计范围，不需要修改核心函数逻辑就能适配不同区间的质数统计需求
• 后续的结果对比、差异计算逻辑可以全量复用，不需要为每个方法单独写对比代码。

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