用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

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