python定义函数求素数

可以使用以下代码定义一个求素数的函数：

def is_prime(num):
    """
    判断一个数是否为素数
    """
    if num < 2:
        return False
    for i in range(2, int(num ** 0.5) + 1):
        if num % i == 0:
            return False
    return True

上面的代码中，我们定义了一个名为 `is_prime` 的函数，它接受一个参数 `num`，用于判断该数是否为素数。如果该数小于 2，则直接返回 False，因为小于 2 的数都不是素数。否则，我们使用一个循环从 2 到该数的平方根进行遍历，如果该数能被其中的任意一个数整除，则说明不是素数，返回 False。如果循环结束后都没有返回 False，就说明该数是素数，返回 True。

这个函数可以用来判断任意一个数是否为素数，例如：

print(is_prime(7)) # True
print(is_prime(10)) # False

如果你需要求一定范围内的素数，可以再定义一个函数，如下所示：

def get_primes(start, end):
    """
    获取指定范围内的所有素数
    """
    primes = []
    for num in range(start, end + 1):
        if is_prime(num):
            primes.append(num)
    return primes

这个函数接受两个参数 `start` 和 `end`，表示要获取素数的范围。我们定义一个空列表 `primes` 来存储找到的素数，然后使用一个循环遍历该范围内的所有数，如果是素数，则加入到 `primes` 列表中。最后返回 `primes` 列表。

例如，我们可以使用下面的代码来获取 1 到 100 之间的所有素数：

primes = get_primes(1, 100)
print(primes)

输出结果为：

[2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67, 71, 73, 79, 83, 89, 97]