python素数判断函数

在Python中，判断一个数是否为素数的函数通常是通过检查这个数是否只有两个不同的正除数：1和它本身。这可以通过遍历从2到该数的平方根的所有整数来完成，因为如果一个数不是素数，它必定有一个因子不大于它的平方根。以下是一个简单的素数判断函数示例：

import math

def is_prime(number):
    if number <= 1:
        return False  # 0和1不是素数
    if number <= 3:
        return True  # 2和3是素数
    if number % 2 == 0 or number % 3 == 0:
        return False  # 排除能被2和3整除的数

    i = 5
    while i * i <= number:
        if number % i == 0 or number % (i + 2) == 0:
            return False
        i += 6

    return True

这个函数首先处理了一些特殊情况，比如小于等于1的数不是素数，2和3是素数。接着，它使用了6k±1优化来检查因子，这是因为除了2和3之外的所有素数都可以表示为6k±1的形式。