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
# 測試
num = 17
if is_prime(num):
print(f"{num} is a prime number.")
else:
print(f"{num} is not a prime number.")
在這個示例中，is_prime函數用來判斷一個數是否為素數。函數首先判斷如果數字小于2，則返回False，因為素數定義為大于1的自然數。然后通過遍歷從2到其平方根的整數來判斷是否有因子，如果有因子則返回False，否則返回True。最后通過測試數字17來驗證函數的正確性。
