下面是一种在 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
print(is_prime(2)) # True
print(is_prime(3)) # True
print(is_prime(4)) # False
该代码通过在 2 到 $\sqrt{n}$ 的范围内