写一个判断是否是素数的Python代码

以下是一个判断是否是素数的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

该函数接受一个整数参数n，如果n是素数则返回True，否则返回False。