使用 joblib 检查 integer 是否为质数。 如果是，计算它的平方； 否则，append None 返回值

Question

import joblib

def is_prime(n):
    for i in range(2, n):
        if n % i == 0:
            return False
    return True

def square_if_prime(n):
    if is_prime(n):
        return n * n
    return None

primes = joblib.Parallel(n_jobs=4)(joblib.delayed(square_if_prime)(n) for n in range(2, 100000))
primes = [prime for prime in primes if prime]
print(primes[:10])

Answer 1

try this:

def is_prime(n):
    for i in range(2,int(n**0.5)+1):
        if n%i==0:
            return False
    return True