创建一个判断一个数是否为质数的程序会有麻烦

import math

def is_prime(n):
    if n <= 1:
        return False
    if n <= 3:
        return True
    if n % 2 == 0 or n % 3 == 0:
        return False
    for i in range(5, int(math.sqrt(n)) + 1, 6):
        if n % i == 0 or n % (i + 2) == 0:
            return False
    return True

# 示例使用方法
num = 17
if is_prime(num):
    print(f"{num} 是质数")
else:
    print(f"{num} 不是质数")