Question

我想遍历python中的列表，检测素数，然后将它们添加到另一个列表中。

primes = []
nprimes = []
for j in range(0, len(arr)):
    num = arr[j] #my list with numbers to check 

    if num > 1:
        for k in range(2, num):
            if (num % k) == 0:
                nprimes.append(num)
                break
            else:
                primes.append(num)
               
    else:
        print(num, " can't be checked, because its smaller than 1")

我有一个问题，总是添加不是质数的数字。同样，一般而言，代码似乎无法正常工作。

Answer 1

如果num % k == 0为假，则不能直接说它为prime，而必须等待整个循环，因此将else与for循环一起使用，它将被执行没有遇到break，这意味着它是素数

您可以直接迭代值for num in arr
您可以在sqrt(num)处停止循环，然后再找不到新的除数

for num in arr:
    if num > 1:
        for k in range(2, int(num ** 0.5) + 1):
            if num % k == 0:
                nprimes.append(num)
                break
        else:
            primes.append(num)
    else:
        print(num, " can't be checked, because its smaller than 1")

Answer 2

出于学习目的，让我们尝试另一种方法。首先，我建议您将审判部门代码移至其自己的谓词函数is_prime()中，该函数返回True或False，因此可以独立于其余代码进行优化。

然后，我要itertools.groupby将列表分为素数序列和非素数序列，然后将它们拼接到适当的列表上：

def is_prime(number):
    if number < 2:
        return False

    if number % 2 == 0:
        return number == 2

    for divisor in range(3, int(number ** 0.5) + 1, 2):
        if number % divisor == 0:
            return False

    return True

if __name__ == "__main__":

    from random import sample
    from itertools import groupby

    array = sample(range(1, 100), 15)

    primes = []
    composites = []

    for are_prime, numbers in groupby(array, is_prime):
        if are_prime:
            primes.extend(numbers)
        else:
            composites.extend(numbers)

    print("Numbers:", array)
    print("Primes:", primes)
    print("Composites:", composites)

输出

% python3 test.py
Numbers: [91, 87, 10, 2, 11, 24, 21, 12, 46, 61, 15, 32, 57, 22, 5]
Primes: [2, 11, 61, 5]
Composites: [91, 87, 10, 24, 21, 12, 46, 15, 32, 57, 22]
%

有很多方法可以解决这个问题，其中许多是有教育意义的！

遍历列表并找到素数并添加到另一个列表

2 个答案: