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` 不大于 1，则返回 `False`。否则，使用一个循环从 2 到 `int(n**0.5)+1`（`int(n**0.5)` 表示 `n` 的平方根取整），依次判断 `n` 能否被这些数整除。如果能，则 `n` 不是质数，返回 `False`。如果循环结束后都没有返回 `False`，则 `n` 是质数，返回 `True`。

可以在其他代码中导入该函数并调用，例如：

from prime import is_prime

n = 17
if is_prime(n):
    print(f"{n}是质数")
else:
    print(f"{n}不是质数")