python编写一个判断素数的函数

以下是一个判断素数的Python函数：

def is_prime(number):
    if number <= 1:
        return False
    if number <= 3:
        return True
    if number % 2 == 0 or number % 3 == 0:
        return False
    i = 5
    while i * i <= number:
        if number % i == 0 or number % (i + 2) == 0:
            return False
        i += 6
    return True

您可以调用这个函数，并将待判断的数字作为参数传递给它。如果函数返回 True，则该数字是素数；如果返回 False，则不是素数。

例如：

number = 17
if is_prime(number):
    print(f"{number} 是素数")
else:
    print(f"{number} 不是素数")