檢查列表中的數字是否為質數

Question

我寫了下面的代碼，它應該檢查列表中的數字是否是質數，但是有一個問題我無法解決，因為我正在嘗試實現檢查到平方根的優化的數字，我有一個TypeError 。

def is_prime(x):
    if x <= 1:
        return False
    if x == 2:
        return True
    for n in range(3, x**(0.5)+1, 2): # this skips even numbers and only checks up to sqrt(x)
        if x % n == 0:
            return False
    return True
    
my_list = [1,2,4,5,6,7]
result = list(map(is_prime,my_list))
print(result)

---

  File "PrimeCheck.py", line 39, in <module>
    result = list(map(is_prime,my_list))
  File "PrimeCheck.py", line 32, in is_prime
    for n in range(3, x**(0.5)+1, 2): # this skips even numbers and only checks up to sqrt(x)
TypeError: 'float' object cannot be interpreted as an integer

Answer 1

x**(0.5)+1不是整數，因此range無法生成列表。

嘗試四舍五入：

from math import ceil
def is_prime(x):

    if x <= 1:
        return False
    if x == 2:
        return True
    for n in range(3, ceil(x**(0.5)), 2):   # this skips even numbers and only checks up to sqrt(x)
        if x % n == 0:
            return False
    return True


my_list = [1,2,4,5,6,7]
result = list(map(is_prime,my_list))

print(result)

Answer 2

表達方式：

range(3, x**(0.5)+1, 2):

提升為浮點類型，因為

x**(0.5)

對於您想要的行為，您需要再次將其設為 int，請嘗試：

range(3, int(round(x**(0.5)+1)), 2):

在哪里

int(x)

告訴 python 你想將 x 解釋為整數類型

和

round(x)

返回最接近 x 的整數。

請注意，舍入可能不是您所期望的：

https://docs.python.org/2/library/functions.html#round

Answer 3

檢查列表中的數字，如果該數字是素數並將素數附加到列表中。