Python 判断素数函数实现代码

以下是一个判断素数的函数的实现代码：

def is_prime(n):
    if n <= 1:
        return False
    for i in range(2, int(n**0.5) + 1):
        if n % i == 0:
            return False
    return True

num = int(input('请输入一个整数：'))
if is_prime(num):
    print(num, '是素数')
else:
    print(num, '不是素数')

在上述代码中，is_prime 函数用于判断一个整数是否为素数。如果输入的整数小于等于 1，则直接返回 False，因为素数定义上是大于 1 的自然数。然后使用一个循环从 2 遍历到 n 的平方根加 1，判断 n 是否能被这些数整除，如果能整除，则返回 False，表示不是素数。如果循环结束后仍然没有找到能整除 n 的数，则说明 n 是素数，返回 True。

在主函数中，首先通过 input 函数获取一个整数输入，并将其转换为整型。然后调用 is_prime 函数判断该整数是否为素数，根据返回值输出相应的结果。