写一段代码判断一个数是否为素数

这是一个简单的 Python 代码片段，用于判断一个数是否为素数：

def is_prime(n):
    if n <= 1:  # 1不是素数，0和负数也不算
        return False
    elif n == 2:  # 2是最小的素数
        return True
    elif n % 2 == 0:  # 偶数除了2以外都不是素数
        return False
    else:  # 对于其他奇数，从3开始检查到n的平方根
        for i in range(3, int(n**0.5) + 1, 2):
            if n % i == 0:
                return False
        return True

# 测试函数
number = int(input("请输入一个整数: "))
if is_prime(number):
    print(f"{number}是素数")
else:
    print(f"{number}不是素数")

这段代码首先排除了小于等于1的数，然后处理特殊情况如2，接着只检查奇数作为潜在因子，并在找到因子时返回False，如果遍历完所有可能的因子都没有找到，则说明是素数。