Python素數檢查

Question

伙計們！

我有一個代碼學院的Python課程練習，其中我必須檢查傳遞的參數是否是素數。 我的代碼如下所示：

def is_prime(x):
    for n in range(2, x-1):
        if (x < 2):
            return False
        elif(x % n == 0):
            return True
        else:
            return False

在我看來，我已經涵蓋了所有可能性，但同時顯示此錯誤：

Your function fails on is_prime(0). It returns None when it should return False.

據我所知，這屬於第一個if條件。 有人可以解釋這怎么可能嗎？

Answer 1

提示，當x為0時range(2, x-1)在range(2, x-1)內的數字是多少？

Answer 2

此版本可能有效。

def is_prime(x):
    if (x < 2): 
        return False
    for n in range(2, x-1): 
        if(x % n == 0): 
            return False
    return True